bzoj 4373 算术天才⑨与等差数列——线段树+set

题目：https://www.lydsy.com/JudgeOnline/problem.php?id=4373

能形成公差为k的等差数列的条件：mx-mn=k*(r-l) && 差分数组gcd为k  && 区间内没有重复的数。

这些都可以线段树维护。

那个“没有重复的数”需要给每个位置记下权值的pre，修改的时候需要找到它前面一个和后面一个。

用链表的话没法插入。可以给每个权值开一个set，就能用lower_bound了。

有些细节。重载运算符很方便。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<map>
#include<set>
using namespace std;
const int N=3e5+,INF=1e9+;
int n,m,c[N],ls[N<<],rs[N<<],tot,pre[N],lt,cnt;
map<int,int> mp;
set<int> s[N<<];
int gcd(int a,int b){return b?gcd(b,a%b):a;}
int abss(int k){return k<?-k:k;}
struct Node{
  int mx,mn,lst,g;
  Node(int a=,int b=,int l=,int g=):mx(a),mn(b),lst(l),g(g) {}
  Node operator+ (const Node &b)const
  {
    return Node(max(mx,b.mx),min(mn,b.mn),max(lst,b.lst),gcd(g,b.g));
  }
}a[N<<];
void add(int x)
{
  mp[x]=++cnt;s[cnt].insert();s[cnt].insert(n+);
}
void pshp(int cr)
{
  a[cr]=a[ls[cr]]+a[rs[cr]];
}
void build(int l,int r,int cr)
{
  if(l==r)
    {
      a[cr]=Node(c[l],c[l],pre[l],abss(c[l+]-c[l]));return;
    }
  int mid=((l+r)>>);
  ls[cr]=++tot;build(l,mid,ls[cr]);
  rs[cr]=++tot;build(mid+,r,rs[cr]);
  pshp(cr);//!!!
}
void updt(int l,int r,int cr,int p)
{
  if(l==r)
    {
      a[cr]=Node(c[l],c[l],pre[l],abss(c[l+]-c[l]));return;
    }
  int mid=((l+r)>>);
  if(p<=mid)updt(l,mid,ls[cr],p);
  else updt(mid+,r,rs[cr],p);
  pshp(cr);
}
Node query(int l,int r,int cr,int L,int R)
{
  if(l>=L&&r<=R)return a[cr];
  int mid=((l+r)>>);
  if(L>mid)return query(mid+,r,rs[cr],L,R);
  if(R<=mid)return query(l,mid,ls[cr],L,R);
  return query(l,mid,ls[cr],L,R)+query(mid+,r,rs[cr],L,R);
}
int main()
{
  scanf("%d%d",&n,&m);
  for(int i=;i<=n;i++)
    {
      scanf("%d",&c[i]);
      if(!mp[c[i]])add(c[i]);
      int t=mp[c[i]];
      pre[i]=*--s[t].lower_bound(i);
      s[t].insert(i);
    }
  a[n+]=a[n];//?是为了让最后一个点的差分是0，使不影响吧
  tot=;build(,n,);
  int op,x,y;
  while(m--)
    {
      scanf("%d",&op);
      if(op==)
    {
      scanf("%d%d",&x,&y);x^=lt;y^=lt;
      int t=mp[c[x]];
      int u=*--s[t].lower_bound(x),v=*s[t].upper_bound(x);
      if(v!=n+)pre[v]=u,updt(,n,,v);//v!=n+1
      s[t].erase(x);
      if(!mp[y])add(y);t=mp[y];
      u=*--s[t].lower_bound(x);v=*s[t].upper_bound(x);
      if(v!=n+)pre[v]=x,updt(,n,,v);//v!=n+1
      pre[x]=u;c[x]=y;s[t].insert(x);
      updt(,n,,x);
      if(x!=)updt(,n,,x-);//x!=1//为了改差分
    }
      else
    {
      int L,R,k;
      scanf("%d%d%d",&L,&R,&k);L^=lt;R^=lt;k^=lt;
      if(L==R){printf("Yes\n");lt++;continue;}//放在下面那个if的上面！
      Node ans0=query(,n,,L,R-),ans=ans0+query(,n,,R,R);//为了差分弄ans0
      if(!k)    //放在下面那个if的上面！
        {
          if(ans.mx==ans.mn)printf("Yes\n"),lt++;
          else printf("No\n");
          continue;
        }
      if(ans.lst<L&&ans.mx-ans.mn==k*(R-L)&&ans0.g==k)printf("Yes\n"),lt++;
      else printf("No\n");
    }
    }
  return ;
}