題目鏈接:點擊查看
題目大意:給出一個長度為 n 的數列,再給出 m 次操作,每次操作分為兩種情況:
C l r d:區間 [ l , r ] 內的數字都加上 d?Q l r :查詢 [ l , r ] 內的區間和
題目分析:分塊的做法比較直觀,將長度為 n 的數列分為 sqrt( n ) 個長度為 sqrt( n ) 的獨立部分,對于查詢和修改:
如果 l 和 r 位于同一個部分內,則直接遍歷即可如果 l 和 r 位于不同的部分內:記 ll 和 rr 分別為第幾個部分 遍歷 [ ll + 1 , rr - 1 ] 的每個部分,給 add 數組加上 val * ( 每個部分的長度?)對于兩頭多出的部分,直接遍歷即可
時間復雜度 ( n + m ) * sqrt( n )
代碼:
?
#include<iostream>
#include<cstdio>
#include<string>
#include<ctime>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<stack>
#include<climits>
#include<queue>
#include<map>
#include<set>
#include<sstream>
#include<cassert>
using namespace std;typedef long long LL;typedef unsigned long long ull;const int inf=0x3f3f3f3f;const int N=1e5+100;int n,m,t,L[N],R[N],pos[N];LL a[N],sum[N],add[N];void update(int l,int r,int val)
{int ll=pos[l],rr=pos[r];if(ll==rr){for(int i=l;i<=r;i++)a[i]+=val;sum[ll]+=val*(r-l+1);}else{for(int i=ll+1;i<=rr-1;i++)add[i]+=val;for(int i=l;i<=R[ll];i++)a[i]+=val;sum[ll]+=(R[ll]-l+1)*val;for(int i=L[rr];i<=r;i++)a[i]+=val;sum[rr]+=(r-L[rr]+1)*val;}
}LL query(int l,int r)
{int ll=pos[l],rr=pos[r];LL ans=0;if(ll==rr){for(int i=l;i<=r;i++)ans+=a[i];ans+=(r-l+1)*add[ll];}else{for(int i=ll+1;i<=rr-1;i++)ans+=sum[i]+add[i]*(R[i]-L[i]+1);for(int i=l;i<=R[ll];i++)ans+=a[i];ans+=(R[ll]-l+1)*add[ll];for(int i=L[rr];i<=r;i++)ans+=a[i];ans+=(r-L[rr]+1)*add[rr];}return ans;
}int main()
{
#ifndef ONLINE_JUDGE
// freopen("input.txt","r",stdin);
// freopen("output.txt","w",stdout);
#endif
// ios::sync_with_stdio(false);scanf("%d%d",&n,&m);for(int i=1;i<=n;i++)scanf("%lld",a+i);t=sqrt(n);for(int i=1;i<=t;i++){L[i]=(i-1)*sqrt(n)+1;R[i]=i*sqrt(n);}if(R[t]<n){t++;L[t]=R[t-1]+1;R[t]=n;}for(int i=1;i<=t;i++)for(int j=L[i];j<=R[i];j++){pos[j]=i;sum[i]+=a[j];}while(m--){char op[5];scanf("%s",op);if(op[0]=='C'){int l,r,val;scanf("%d%d%d",&l,&r,&val);update(l,r,val);}else{int l,r;scanf("%d%d",&l,&r);printf("%lld\n",query(l,r));}}return 0;
}
?
總結
以上是生活随笔為你收集整理的POJ - 3468 A Simple Problem with Integers(分块)的全部內容,希望文章能夠幫你解決所遇到的問題。
如果覺得生活随笔網站內容還不錯,歡迎將生活随笔推薦給好友。