1. 程式人生 > >BZOJ 4373(算術天才⑨與等差數列-線段樹+hash)

BZOJ 4373(算術天才⑨與等差數列-線段樹+hash)

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

我們暴力維護區間最大最小值,區間和,區間平方和
詢問時判斷一個區間的首項末項,公差,和,平方和(這個要用等差數列平方和公式)

#include<bits/stdc++.h>
using namespace std;
#define For(i,n) for(int i=1;i<=n;i++)
#define Fork(i,k,n) for(int i=k;i<=n;i++)
#define Rep(i,n) for(int i=0;i<n;i++) #define ForD(i,n) for(int i=n;i;i--) #define ForkD(i,k,n) for(int i=n;i>=k;i--) #define RepD(i,n) for(int i=n;i>=0;i--) #define Forp(x) for(int p=Pre[x];p;p=Next[p]) #define Forpiter(x) for(int &p=iter[x];p;p=Next[p]) #define Lson (o<<1) #define
Rson ((o<<1)+1)
#define MEM(a) memset(a,0,sizeof(a)); #define MEMI(a) memset(a,0x3f,sizeof(a)); #define MEMi(a) memset(a,128,sizeof(a)); #define MEMx(a,b) memset(a,b,sizeof(a)); #define INF (0x3f3f3f3f) #define F (1000000007) #define pb push_back #define mp make_pair #define fi first #define se second #define vi vector<int> #define pi pair<int,int> #define SI(a) ((a).size()) #define Pr(kcase) printf("Case %d:\n",kcase); #define PRi(a,n) For(i,n-1) cout<<a[i]<<' '; cout<<a[n]<<endl; #define PRi2D(a,n,m) For(i,n) { \ For(j,m-1) cout<<a[i][j]<<' ';\ cout<<a[i][m]<<endl; \ } #pragma comment(linker, "/STACK:102400000,102400000") typedef long double ld; typedef unsigned long long ll; ll mul(ll a,ll b){return (a*b)%F;} ll add(ll a,ll b){return (a+b)%F;} ll sub(ll a,ll b){return ((a-b)%F+F)%F;} void upd(ll &a,ll b){a=(a%F+b%F)%F;} int read() { int x=0,f=1; char ch=getchar(); while(!isdigit(ch)) {if (ch=='-') f=-1; ch=getchar();} while(isdigit(ch)) { x=x*10+ch-'0'; ch=getchar();} return x*f; } const int maxn =300010; ll sum[maxn<<2],minv[maxn<<2],maxv[maxn<<2],sum2[maxn<<2],a[maxn]; void pushup(int o) { sum[o]=sum[Lson]+sum[Rson]; sum2[o]=sum2[Lson]+sum2[Rson]; minv[o]=min(minv[Lson],minv[Rson]); maxv[o]=max(maxv[Lson],maxv[Rson]); } void build(int l,int r,int o) { if (l==r) { sum[o]=minv[o]=maxv[o]=a[l]; sum2[o]=(ll)a[l]*a[l]; return ; } int m=(l+r)>>1; build(l,m,Lson),build(m+1,r,Rson); pushup(o); } void update(int l,int r,int o,int p,ll v) { if (l==r) { sum[o]=minv[o]=maxv[o]=v; sum2[o]=(ll)v*v; return ; } int m=(l+r)>>1; if (p<=m) update(l,m,Lson,p,v); else update(m+1,r,Rson,p,v); pushup(o); } ll ret1,ret2,ret3,ret4; void query(int l,int r,int o,int L,int R) { if(L<=l && r<=R ) { ret1+=sum[o]; ret2=min(ret2,minv[o]); ret3=max(ret3,maxv[o]); ret4+=sum2[o]; return ; } int m=(l+r)>>1; if(L<=m) { query(l,m,Lson,L,R); } if(m<R) { query(m+1,r,Rson,L,R); } } int main() { // freopen("bzoj4373.in","r",stdin); // freopen(".out","w",stdout); int n=read(),m=read(); For(i,n) a[i]=read(); build(1,n,1); ll ans=0; while(m--) { ll op=read(), p1=read()^ans,p2=read()^ans; if (op==1) { update(1,n,1,p1,p2); } else { ll p3=read()^ans; ret1=0,ret2=INF,ret3=0,ret4=0; query(1,n,1,p1,p2); bool flag; if (p1==p2) flag=1; else if ((ret3-ret2)%(p2-p1)) flag=0; else { ll k2=(ret3-ret2)/(p2-p1); if (k2^p3) flag=0; else { ll len=p2-p1+1; ll su1=(ret2+ret3)*len/2,su2=len*ret2*ret3+(len*(len-1)*(2*len-1) )/6* p3* p3; flag= su2==ret4 && su1==ret1; } } puts(flag?"Yes":"No"); ans+=flag; } } return 0; }