C - 东风不与周郎便
Time Limit: 3000/1000MS (Java/Others) Memory Limit: 65535/65535KB (Java/Others)
“揽二乔于东南兮,乐朝夕之与共”
一首铜雀台赋,将愤怒与恐惧散播在了孙吴大军之中。
对抗曹军,万事俱备,只欠东风。
现在已经找到n个风眼,这些风眼的东风有强有弱,诸葛亮说他每次祈风都能够将一段风眼的东风增强,但需人去帮他布阵。同时他需要时刻掌控风眼的状况,以确定下一步的计划,所以还需要知道一段风眼的强度之和。
借东风,此乃逆天之术,施术者会折阳寿不说,布阵者更是会受十倍之伤。
“何人能当此任?”
“在下愿往,大都督。”
“你是?”
“在下一无名小卒,来自跳蚤街。”
Input
第一行两个整数n,m,表示有n个风眼,诸葛亮一共祈风或询问了m次。
第二行n个整数,第i个数ai表示第i个风眼已有东风的强度。
接下来m行,每行开始先读入一个整数si,指明这是一次询问还是一次祈风。
si=0,表明这是一次询问,然后读入两个整数li,ri,表示询问[li,ri]区间中风眼的东风强度之和。
si=1,表明这是一次祈风,然后读入三个整数li,ri,wi,表示把[li,ri]区间中每个风眼的东风强度提升wi。
1≤n,m≤100000,0≤ai≤10000,0≤wi≤10000,1≤li≤ri≤n
Output
有多少询问就输出多少行,每行输出一个整数,作为对该询问的回答。
Sample input and output
Sample Input | Sample Output |
---|---|
5 4 1 2 3 4 5 1 2 3 2 0 3 4 1 4 5 3 0 2 4 |
9 16 |
这道题是B题的进阶版,需要修改区间。
我是这样做的,在每个区间加一个标记,如果该区间所有元素都在区间修改中给覆盖,就给该标记加上修改的值,在询问区间时,如果某区间有标记,则给答案加上询问区间与该区间交集长度*标记大小。如此就可AC。
#include <iostream> #include<stdio.h> #include<string.h> using namespace std; long long q[100005],shu[400005][4]; void jianshu(long long g,long long h,long long l) { shu[l][1]=g;shu[l][2]=h; if (g==h) {shu[l][0]=q[g];return;} jianshu(g,(g+h)/2,l*2); jianshu((g+h)/2+1,h,l*2+1); } long long suan(long long j) { if (shu[j][1]==shu[j][2]) return shu[j][0]; return shu[j][0]=suan(j*2)+suan(j*2+1); } long long found(long long l,long long g,long long h) { if (g==shu[l][1] && h==shu[l][2]) return shu[l][0]; long long z=(shu[l][1]+shu[l][2])/2; if (g<=z && h>=z+1) return found(l*2,g,z)+found(l*2+1,z+1,h)+shu[l][3]*(h-g+1); if (h<=z) return found(l*2,g,h)+shu[l][3]*(h-g+1); else return found(l*2+1,g,h)+shu[l][3]*(h-g+1); } void done(long long l,long long g,long long h,long long k) { long long z; shu[l][0]+=k*(h-g+1); if (shu[l][1]==shu[l][2]) return; if (shu[l][1]==g && shu[l][2]==h) {shu[l][3]+=k;return;} z=(shu[l][1]+shu[l][2])/2; if (g<=z && h>=z+1) {done(l*2,g,z,k);done(l*2+1,z+1,h,k);return ;} if (h<=z) done(l*2,g,h,k); else done(l*2+1,g,h,k); } int main() { long long n,m,j,l,g,h,k; scanf("%lld%lld",&n,&m); memset(q,0,sizeof(q)); memset(shu,0,sizeof(shu)); for (j=1;j<=n;j++) scanf("%lld",&q[j]); jianshu(1,n,1); suan(1); for (j=1;j<=m;j++){ scanf("%lld %lld %lld",&l,&g,&h); if (l==0) printf("%lld\n",found(1,g,h)); if (l==1) {scanf("%lld",&k);done(1,g,h,k);} } return 0; }