BZOJ 4373 算术天才⑨与等差数列 线段树+set（恶心死我了）

mdzz，这道题重构了4遍，花了一个晚上。。。

---

满足等差数列的条件：

1. 假设min是区间最小值，max是区间最大值，那么 max-min+k(r−l)

2. 区间相邻两个数之差的绝对值的gcd=k

3. 区间没有重复的数

前两个条件直接线段树就好啦；而第三个条件：对于每个权值开个set，值为位置（离散化）然后维护一个pp[i]，表示当前a[i]这个值，在i前面最后一次出现的位置。

那么满足第3个条件，当且仅当区间[l,r]的 max { pre[ i ] } ( l <= i <= r ) 小于l，这个也是用线段树维护。

最后看修改操作：在set上找前一个数，后一个数，然后修改相应的pre值。

#include<cstdio>
#include<iostream>
#include<map>
#include<set>
#define R register int
#define pc(x) putchar(x)
#define ls (tr<<1)
#define rs (tr<<1|1)
using namespace std;
const int N=,M=;
int n,m,cnt,num,mod,G,mmn,mmx;
int mn[M],mx[M],pp[M],gg[M],a[N],b[N],pos[N],d[N];
bool flg;
map<int,int>mp;
set<int>s[N<<];
set<int>::iterator pre,nxt;
inline int g() {
    R ret=; register char ch; while(!isdigit(ch=getchar())) ;
    do ret=ret*+(ch^); while(isdigit(ch=getchar())); return ret;
}
inline int abs(int a) {return a>?a:-a;}
inline int gcd(int a,int b) {return b?gcd(b,a%b):a;}
inline int calc(int x) { R ret;
    if(mp.find(x)==mp.end()) ret=mp[x]=++cnt; else return mp[x];
    s[ret].insert(),s[ret].insert(n+); return ret;
}
inline void build(int tr,int l,int r) {
    if(l==r) {mn[tr]=mx[tr]=a[l]; pp[tr]=pos[l]; gg[tr]=d[l]; return ;}
    R md=(l+r)>>; build(ls,l,md),build(rs,md+,r);
    mn[tr]=min(mn[ls],mn[rs]),mx[tr]=max(mx[ls],mx[rs]);
    pp[tr]=max(pp[ls],pp[rs]),gg[tr]=gcd(gg[ls],gg[rs]);
}
inline void updatep(int tr,int l,int r,int pos,int p) {
    if(l==r) {pp[tr]=p; return ;} R md=(l+r)>>;
    if(pos<=md) updatep(ls,l,md,pos,p); else updatep(rs,md+,r,pos,p);
    pp[tr]=max(pp[ls],pp[rs]);
}
inline void updateg(int tr,int l,int r,int pos,int G) {
    if(l==r) {gg[tr]=G; return;} R md=(l+r)>>;
    if(pos<=md) updateg(ls,l,md,pos,G); else updateg(rs,md+,r,pos,G);
    gg[tr]=gcd(gg[ls],gg[rs]);
}
inline void update(int tr,int l,int r,int pos,int inc,int pre) {
    if(l==r) {mn[tr]=mx[tr]=inc; pp[tr]=pre; return ;} R md=(l+r)>>;
    if(pos<=md) update(ls,l,md,pos,inc,pre); else update(rs,md+,r,pos,inc,pre);
    mn[tr]=min(mn[ls],mn[rs]),mx[tr]=max(mx[ls],mx[rs]),pp[tr]=max(pp[ls],pp[rs]);
}
inline void ask(int tr,int l,int r,int LL,int RR) {
    if(LL<=l&&r<=RR) {mmn=min(mmn,mn[tr]),mmx=max(mmx,mx[tr]); if(pp[tr]>=LL) flg=; return;}
    R md=(l+r)>>; if(LL<=md) ask(ls,l,md,LL,RR); if(RR>md) ask(rs,md+,r,LL,RR);
}
inline void askg(int tr,int l,int r,int LL,int RR) {
    if(LL<=l&&r<=RR) {G=gcd(gg[tr],G); return ;} R md=(l+r)>>;
    if(LL<=md) askg(ls,l,md,LL,RR); if(RR>md) askg(rs,md+,r,LL,RR);
}
inline bool judge(int l,int r) {
    if(l==r) return true; flg=false,mmn=0x3f3f3f3f,mmx=G=;
    ask(,,n,l,r); if(!mod) return mmx==mmn;
    if(flg) return false;
    if(mmx-mmn!=1ll*(r-l)*mod) return false;
    askg(,,n,l,r-); return G%mod==;
}
signed main() {
    n=g(),m=g();
    for(R i=;i<=n;i++) {
        a[i]=g(); s[b[i]=calc(a[i])].insert(i);
        pre=s[b[i]].find(i); pos[i]=*(--pre);
    } for(R i=;i<n;i++) d[i]=abs(a[i]-a[i+]); build(,,n);
    for(R i=;i<=m;++i) { R k=g(),l=g()^num,r=g()^num;
        if(k&) {
            pre=nxt=s[b[l]].find(l); pre--,nxt++;
            if(*nxt<n) updatep(,,n,*nxt,*pre);
            s[b[l]].erase(l),b[l]=calc(r),s[b[l]].insert(l);
            pre=nxt=s[b[l]].find(l); pre--,nxt++;
            if(*nxt<n) updatep(,,n,*nxt,l); update(,,n,l,a[l]=r,*pre);
            if(l>) updateg(,,n,l-,abs(a[l-]-r));
            if(l<n) updateg(,,n,l,abs(r-a[l+]));
        } else mod=g()^num,judge(l,r)?(++num,pc('Y'),pc('e'),pc('s')):(pc('N'),pc('o')),pc('\n');
    }
}

---

2019.04.22 这道题让我想扔掉OI。。。气死我了。。。QAQ