loj #6138. 「2017 山东三轮集训 Day4」Right

题目:

题解:

暴力一波 \(SG\) 函数可以发现这么一个规律:

• \(p\) 为奇数的时候 : \(SG(n) = n \% 2\)
• \(p\) 为偶数的时候 : \(SG(n) = n \% (p+1) == p ? 2 : n \% (p+1) \% 2\)

对于奇数的情况我们就可以直接用一棵支持区间取反和区间查询 \(1\) 的个数的线段树搞定。
那么难点在于偶数的情况。

我们可以采用分块算法.
每个块分别中保存 \(\bmod (p+1)\) 为奇数的数和为偶数的数。
然后每次查询的时候我们可以通过几次 \(lower_bound\) 来解决。
复杂度 \(O(m\log n)\space or\space O(m\sqrt{n}\log n)\)

#include <cmath>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long ll;
typedef pair<int*,int*> pa;
inline void read(int &x){
    x=0;static char ch;static bool flag;flag = false;
    while(ch=getchar(),ch<'!');if(ch == '-') ch=getchar(),flag = true;
    while(x=10*x+ch-'0',ch=getchar(),ch>'!');if(flag) x=-x;
}
#define rg register int
#define rep(i,a,b) for(rg i=(a);i<=(b);++i)
#define per(i,a,b) for(rg i=(a);i>=(b);--i)
const int maxn = 100010;
int n,q,p;
namespace work1{
    int T[maxn<<2],tag[maxn<<2],a[maxn];
    inline void push_down(int rt,int l,int r){
        if(l == r || tag[rt] == 0) return ;
        int mid = l+r >> 1;
        tag[rt<<1] ^= 1;
        if((mid-l+1)&1) T[rt<<1] ^= 1;
        tag[rt<<1|1] ^= 1;
        if((r-mid)&1) T[rt<<1|1] ^= 1;
        tag[rt] = 0;
    }
    void build(int rt,int l,int r){
        if(l == r){
            T[rt] = a[l]&1;
            return ;
        }int mid = l+r >> 1;
        build(rt<<1,l,mid);build(rt<<1|1,mid+1,r);
        T[rt] = T[rt<<1]+T[rt<<1|1] & 1;
    }
    int L,R;
    void modify(int rt,int l,int r){
        if(L <= l && r <= R){
            tag[rt] ^= 1;
            if((r-l+1)&1) T[rt] ^= 1;
            return ;
        }int mid = l+r >> 1;push_down(rt,l,r);
        if(L <= mid) modify(rt<<1,l,mid);
        if(R >  mid) modify(rt<<1|1,mid+1,r);
        T[rt] = T[rt<<1]+T[rt<<1|1] & 1;
    }
    int query(int rt,int l,int r){
        if(L <= l && r <= R) return T[rt];
        int mid = l+r >> 1;push_down(rt,l,r);
        if(R <= mid) return query(rt<<1,l,mid);
        if(L >  mid) return query(rt<<1|1,mid+1,r);
        return query(rt<<1,l,mid) + query(rt<<1|1,mid+1,r) & 1;
    }
    int main(){
        rep(i,1,n) read(a[i]);
        build(1,1,n);
        int op,x;
        while(q--){
            read(op);
            if(op == 0){
                read(L);read(R);read(x);
                if(x & 1) modify(1,1,n);
            }else{
                read(L);read(R);
                printf("%d\n",query(1,1,n));
            }
        }
        return 0;
    }
}
namespace work2{
    const int maxm = 322;
    int tmp[maxn],c[maxn];
    struct Node{
        int a[maxm],b[maxm],tag;
        int cnta,cntb,l,r;
        // a[i] % (p+1) = 1;
        // b[i] % (p+1) = 0;
        void build(int n){
            sort(tmp+1,tmp+n+1);
            cnta = cntb = 0;
            rep(i,1,n){
                if(tmp[i] % (p+1) & 1) a[++cnta] = tmp[i];
                else b[++cntb] = tmp[i];
            }return ;
        }
        int query_all(int &one,int &two){
            int x = (p - tag + (p+1)) % (p+1);pa pos;
            if(x & 1) pos = equal_range(a+1,a+cnta+1,x);
            else pos = equal_range(b+1,b+cntb+1,x);
            two += (pos.second - pos.first);
            if(tag <= (p-1)){
                x = p - 1 - tag;int ps = 0;
                if(x & 1) ps = upper_bound(a+1,a+cnta+1,x) - a - 1;
                else ps = upper_bound(b+1,b+cntb+1,x) - b - 1;
                one += ps;
            }
            if(tag >= 2){
                x = p - tag;int ps = 0;
                if(x & 1) ps = cnta - (upper_bound(a+1,a+cnta+1,x)-a-1);
                else ps = cntb - (upper_bound(b+1,b+cntb+1,x)-b-1);
                one += ps;
            }   
        }
        void query(int l,int r,int &one,int &two){
            int cnt = 0;
            rep(i,this->l,this->r){
                c[i] += tag;
                if(c[i] >= p+1) c[i] -= p+1;
                tmp[++cnt] = c[i];
            }sort(tmp+1,tmp+cnt+1);
            cnta = cntb = tag = 0;
            rep(i,1,cnt){
                if(tmp[i] % (p+1) & 1) a[++cnta] = tmp[i];
                else b[++cntb] = tmp[i];
            }
            rep(i,l,r){
                if(c[i] % (p+1) == p) ++ two;
                else if(c[i] % (p+1) & 1) ++ one;
            }return ;
        }
        void modify(int l,int r,int x){
            int cnt = 0;
            rep(i,this->l,this->r){
                c[i] += tag;if(c[i] >= (p+1)) c[i] -= (p+1);
                if(l <= i && i <= r){
                    c[i] += x;
                    if(c[i] >= (p+1)) c[i] -= (p+1);
                }
                tmp[++cnt] = c[i];
            }sort(tmp+1,tmp+cnt+1);
            cnta = cntb = tag = 0;
            rep(i,1,cnt){
                if(tmp[i] % (p+1) & 1) a[++cnta] = tmp[i];
                else b[++cntb] = tmp[i];
            }return ;
        }
    }zs[maxm];
    int belong[maxn];
    inline void modify(int l,int r,int x){
        if(belong[l] == belong[r]) return zs[belong[l]].modify(l,r,x);
        zs[belong[l]].modify(l,zs[belong[l]].r,x);
        zs[belong[r]].modify(zs[belong[r]].l,r,x);
        rep(i,belong[l]+1,belong[r]-1){
            zs[i].tag += x;
            if(zs[i].tag >= (p+1)) zs[i].tag -= (p+1);
        }
        return ;
    }
    inline int query(int l,int r){
        int res1 = 0,res2 = 0;
        if(belong[l] == belong[r]){
            zs[belong[l]].query(l,r,res1,res2);
        }else{
            zs[belong[l]].query(l,zs[belong[l]].r,res1,res2);
            zs[belong[r]].query(zs[belong[r]].l,r,res1,res2);
            rep(i,belong[l]+1,belong[r]-1) zs[i].query_all(res1,res2);
        }
        if((res1 & 1) || (res2 & 1)) return 1;
        else return 0;
    }
    int main(){
        int block = ceil(sqrt(n));
        int sz = 0,m = 1;
        zs[1].l = 1;
        rep(i,1,n){
            belong[i] = m;read(tmp[++sz]);
            c[i] = (tmp[sz] %= (p+1));
            if(sz == block){
                zs[m].build(sz);
                zs[m].r = i,zs[++m].l = i+1;
                sz = 0;
            }
        }
        if(zs[m].l != n+1){
            zs[m].r = n;
            zs[m].build(n-zs[m].l+1);
        }else --m;
        int op,l,r,x;
        while(q--){
            read(op);read(l);read(r);
            if(op == 0) read(x),modify(l,r,x % (p+1));
            else printf("%d\n",query(l,r));
        }
        return 0;
    }
}
int main(){
    read(n);read(q);read(p);
    if(p & 1) work1::main();
    else work2::main();
    return 0;
}

转载于:https://www.cnblogs.com/Skyminer/p/7132867.html