前缀和 + ST表 ---- CF 1556 E. Equilibrium(两个序列 + - 操作使得每位相等) 详解

题目连接


题目大意:

就是给你两个长度为 n n n a , b a,b a,b数组,给你 q ∈ [ 1 , 1 e 5 ] q\in[1,1e5] q[1,1e5]次询问,每次询问一个区间 [ l , r ] [l,r] [l,r]

你对这个区间里面的数可以进行一下操作
取出偶数个位置
l ≤ p o s 1 < p o s 2 < . . . . < p o s z ≤ r ∣ [ z % 2 = = 0 ] l\leq pos_1lpos1<pos2<....<poszr[z%2==0]
对于里面 p o s 1 , p o s 3 , p o s 5 . . . , p o s z − 1 pos_1,pos_3,pos_5...,pos_{z-1} pos1,pos3,pos5...,posz1 a a a数组位置的数 + 1 +1 +1
对于其他位置的 b b b数组的位置 + 1 +1 +1
就是偶数对 b b b操作,奇数对 a a a操作

q u e r y : query: query:最小操作使得 a i = b i ∣ i ∈ [ l , r ] a_i=b_i|i\in[l,r] ai=bii[l,r]
不满足输出 − 1 -1 1


解题思路:

思路特别妙:就是我们先构造 s u b [ i ] = a [ i ] − b [ i ] ∣ i ∈ [ l , r ] sub[i]=a[i]-b[i]|i\in[l,r] sub[i]=a[i]b[i]i[l,r]
如果 a [ i ] + 1 a[i]+1 a[i]+1那么 s u b [ i ] + 1 sub[i]+1 sub[i]+1,如果 b [ i ] + 1 b[i]+1 b[i]+1,那么 s u b [ i ] − 1 sub[i]-1 sub[i]1
那么我们就把 p o s i , p o s j pos_i,pos_j posi,posj看成一对操作分别对 a , b a,b a,b操作那么就是有点像差分了对于 s u b 数 组 sub数组 sub
那么我们考虑最终状态那么就是每个位置 s u b sub sub都是 0 0 0
但是问题转化到这里最小操作还是不好求因为我们知道对于一对操作中间是不予许交叉的,就是必须连续。
但是我们可以知道一个 − 1 -1 1的条件就是 ∑ i = l r s u b [ i ] ! = 0 ∣ + 1 , − 1 和 是 不 会 变 的 \sum_{i=l}^rsub[i]!=0|+1,-1和是不会变的 i=lrsub[i]!=0+11
那么我们可以对 s u b sub sub求一个前缀和 s [ i ] s[i] s[i]:把差分变成连续操作: 区 间 加 区间加
那么:
前缀和 + ST表 ---- CF 1556 E. Equilibrium(两个序列 + - 操作使得每位相等) 详解_第1张图片


AC code

#include 
#define mid ((l + r) >> 1)
#define Lson rt << 1, l , mid
#define Rson rt << 1|1, mid + 1, r
#define ms(a,al) memset(a,al,sizeof(a))
#define log2(a) log(a)/log(2)
#define lowbit(x) ((-x) & x)
#define IOS std::ios::sync_with_stdio(0); cin.tie(0); cout.tie(0)
#define INF 0x3f3f3f3f
#define LLF 0x3f3f3f3f3f3f3f3f
#define f first
#define s second
#define endl '\n'
using namespace std;
const int N = 2e6 + 10, mod = 1e9 + 9;
const int maxn = 2e5 + 10;
const long double eps = 1e-5;
const int EPS = 500 * 500;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> PII;
typedef pair<ll,ll> PLL;
typedef pair<double,double> PDD;
template<typename T> void read(T &x) {
   x = 0;char ch = getchar();ll f = 1;
   while(!isdigit(ch)){if(ch == '-')f*=-1;ch=getchar();}
   while(isdigit(ch)){x = x*10+ch-48;ch=getchar();}x*=f;
}
template<typename T, typename... Args> void read(T &first, Args& ... args) {
   read(first);
   read(args...);
}
int arr[maxn], brr[maxn];
ll pre[maxn];
int lg[maxn]={-1};
ll STmax[30][maxn];
ll STmin[30][maxn];

inline void init(int n) {
    for(int i = 1; i <= lg[n]; ++ i) {
        for(int j = 1; j + (1 << i) - 1 <= n; ++ j) {
            STmax[i][j] = max(STmax[i-1][j],STmax[i-1][j+(1<<(i-1))]);
            STmin[i][j] = min(STmin[i-1][j],STmin[i-1][j+(1<<(i-1))]);
        }
    }
}
inline ll getmax(int l, int r){
    int len = lg[r-l+1];
    return max(STmax[len][l],STmax[len][r-(1<<len)+1]);
}

inline ll getmin(int l, int r){
    int len = lg[r-l+1];
    return min(STmin[len][l],STmin[len][r-(1<<len)+1]);
}

int main() {
    IOS;
    int T;
    // cin >> T;
    T = 1;
    while(T --) {
        int n, m;
        cin >> n >> m;
        for(int i = 1; i <= n; ++ i) cin >> arr[i];
        for(int i = 1; i <= n; ++ i) {
            lg[i] = lg[i/2] + 1;
            int x;
            cin >> x;
            pre[i] = pre[i-1] + arr[i] - x;
            // cout << pre[i] << " ";
            STmax[0][i] = pre[i];
            STmin[0][i] = pre[i];
        }
        // cout << endl;
        init(n);
        // cout << STmax[2][0] << " ";
        // cout << getmax(2,6) << endl;
        while(m --) {
            int l, r;
            cin >> l >> r;
            if(pre[r]-pre[l-1] != 0) cout << "-1\n";
            else if(getmax(l,r) > pre[l-1]) cout << "-1\n";
            else cout << pre[l-1] - getmin(l,r) << "\n";
        }
    }
    return 0;
}

你可能感兴趣的:(数据结构,c语言,r语言,开发语言)