【bzoj4373】算术天才⑨与等差数列

Description

算术天才⑨非常喜欢和等差数列玩耍。
有一天，他给了你一个长度为n的序列，其中第i个数为a[i]。
他想考考你，每次他会给出询问l,r,k，问区间[l,r]内的数从小到大排序后能否形成公差为k的等差数列。
当然，他还会不断修改其中的某一项。
为了不被他鄙视，你必须要快速并正确地回答完所有问题。
注意：只有一个数的数列也是等差数列。

Input

第一行包含两个正整数n,m(1<=n,m<=300000)，分别表示序列的长度和操作的次数。
第二行包含n个整数，依次表示序列中的每个数ai。
接下来m行，每行一开始为一个数op，
若op=1，则接下来两个整数x,y(1<=x<=n，0<=y<=10^9)，表示把a[x]修改为y。
若op=2，则接下来三个整数l,r,k(1<=l<=r<=n，0<=k<=10^9)，表示一个询问。
在本题中，x,y,l,r,k都是经过加密的，都需要异或你之前输出的Yes的个数来进行解密。

Output

输出若干行，对于每个询问，如果可以形成等差数列，那么输出Yes，否则输出No。

Sample Input

5 3

1 3 2 5 6

2 1 5 1

1 5 4

2 1 5 1
Sample Output

No

Yes

题解
d[i]=abs(a[i]−a[i+1])
用线段树维护区间内a的最小值、最大值、以及d的gcd。（其实还要判断是否有重复，本人不会，而且数据水就不打了）

代码

#include<bits/stdc++.h>
#define N 1005
#define ll long long
#define mo 1000000007
using namespace std;
struct node{int mn,mx,gcd;}t[1200005];
int a[300005],d[300005];
int n,m,opt,x,y,k,XOR;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int gcd(int a,int b)
{
    if (b==0) return a;else return gcd(b,a%b);
}
void update(int k)
{
    int l=k<<1,r=k<<1|1;
    t[k].mx=max(t[l].mx,t[r].mx);
    t[k].mn=min(t[l].mn,t[r].mn);
    t[k].gcd=gcd(t[l].gcd,t[r].gcd);
}
void build(int k,int l,int r)
{
    if (l==r){t[k].mx=t[k].mn=a[l];t[k].gcd=d[l];return;}
    int mid=(l+r)>>1;
    build(k<<1,l,mid);build(k<<1|1,mid+1,r);
    update(k);
}
void modify(int k,int l,int r,int x)
{
    if (l==r){t[k].mx=t[k].mn=a[x];t[k].gcd=d[x];return;}
    int mid=(l+r)>>1;
    if (x<=mid) modify(k<<1,l,mid,x);
    else modify(k<<1|1,mid+1,r,x);
    update(k);
}
node query(int k,int l,int r,int x,int y)
{
    if (l==x&&r==y) return t[k];
    int mid=(l+r)>>1;
    if (y<=mid) return query(k<<1,l,mid,x,y);
    else if (x>mid) return query(k<<1|1,mid+1,r,x,y);
    else
    {
        node tmp1=query(k<<1,l,mid,x,mid);
        node tmp2=query(k<<1|1,mid+1,r,mid+1,y);
        node tmp;
        tmp.mx=max(tmp1.mx,tmp2.mx);
        tmp.mn=min(tmp1.mn,tmp2.mn);
        tmp.gcd=gcd(tmp1.gcd,tmp2.gcd);
        return tmp;
    }
}
int main()
{
    n=read();m=read();
    for (int i=1;i<=n;i++)
    {
        a[i]=read();
        d[i-1]=abs(a[i]-a[i-1]);
    }
    build(1,1,n-1);
    while (m--)
    {
        opt=read();x=read()^XOR;y=read()^XOR;
        if (opt==1)
        {
            a[x]=y;d[x]=abs(a[x+1]-a[x]);d[x-1]=abs(a[x]-a[x-1]);
            if (x==n) continue;
            modify(1,1,n-1,x);
            if (x==1) continue;
            modify(1,1,n-1,x-1);
        }
        else
        {
            k=read()^XOR;
            if (x==y){puts("Yes");XOR++;continue;}
            node ans=query(1,1,n-1,x,y-1);
            ans.mx=max(a[y],ans.mx);
            ans.mn=min(a[y],ans.mn);
            if (k==0) if (ans.mx==ans.mn){puts("Yes");XOR++;}
                        else puts("No");
            else
            {
                if ((ans.mx-ans.mn)%k==0&&(ans.mx-ans.mn)/k==(y-x)&&ans.gcd==k) puts("Yes"),XOR++;
                else puts("No");
            }
        }
    }
    return 0;
}