【线段树】bzoj3922 Karin的弹幕
设置一个值K。
d<=K:建立多组线段树;d>K:暴力。
最优时间复杂度的伪计算:
O(n*K*logn(建树)+m*logn(询问类型1)+m*n/K(询问类型2)+m*K*logn(修改))。
求此函数最小值,易得,当K=sqrt(m/logn)时,
时间复杂度:O(m*sqrt(m*logn))。
空间复杂度:O(n*sqrt(m/logn))。
当然,这个计算显然不完全合理,而且,由于使用STL的vector的原因,导致实际建树要慢得多,因此K取得小一些更加合适(跑几组数据自己看看就行了)。如果不稍微小一点是卡不进内存和时间的哦。
#include<cstdio> #include<vector> #include<cmath> using namespace std; #define lson rt<<1,l,m #define rson rt<<1|1,m+1,r #define INF 2147483647 int n,x0,d,a[70001],lim,m; bool op; vector<int>b[70][70],maxv[70][70]; void buildtree(int x,int y,int rt,int l,int r) { if(l==r) { maxv[x][y][rt]=b[x][y][l]; return; } int m=l+r>>1; buildtree(x,y,lson); buildtree(x,y,rson); maxv[x][y][rt]=max(maxv[x][y][rt<<1],maxv[x][y][rt<<1|1]); } void update(int x,int y,int p,int v,int rt,int l,int r) { if(l==r) {maxv[x][y][rt]+=v; return;} int m=l+r>>1; if(p<=m) update(x,y,p,v,lson); else update(x,y,p,v,rson); maxv[x][y][rt]=max(maxv[x][y][rt<<1],maxv[x][y][rt<<1|1]); } int query(int x,int y,int ql,int qr,int rt,int l,int r) { if(ql<=l&&r<=qr) return maxv[x][y][rt]; int m=l+r>>1,res=-INF; if(ql<=m) res=max(res,query(x,y,ql,qr,lson)); if(m<qr) res=max(res,query(x,y,ql,qr,rson)); return res; } int main() { scanf("%d",&n); for(int i=1;i<=n;++i) scanf("%d",&a[i]); scanf("%d",&m); lim=(int)sqrt((double)m/(log((double)n)/log(2.0)))/14; if(!lim) lim=1; for(int i=1;i<=lim;++i)//枚举公差 for(int j=1;j<=lim;++j)//枚举首项 { b[i][j].push_back(0); for(int k=j;k<=n;k+=i) b[i][j].push_back(a[k]); maxv[i][j].assign((b[i][j].size()-1)<<2|1,0); buildtree(i,j,1,1,b[i][j].size()-1); } for(int i=1;i<=m;++i) { scanf("%d%d%d",&op,&x0,&d); if(!op) { a[x0]+=d; for(int j=1;j<=lim;++j)//枚举公差 { int bel=x0%j; if(!bel) bel=j;//计算首项 int pos=x0/j; if(bel!=j) ++pos;//计算pos是该等差数列的第几项 update(j,bel,pos,d,1,1,b[j][bel].size()-1); } } else { if(d>lim) { int res=-INF; for(int j=x0;j<=n;j+=d) res=max(res,a[j]); printf("%d\n",res); } else { int bel=x0%d; if(!bel) bel=d; int sta=x0/d; if(bel!=d) ++sta; printf("%d\n",query(d,bel,sta,b[d][bel].size()-1,1,1,b[d][bel].size()-1)); } } } return 0; }
——The Solution By AutSky_JadeK From UESTC
转载请注明出处:http://www.cnblogs.com/autsky-jadek/