2014 UESTC Training for Data Structures C

C - 东风不与周郎便

Time Limit: 3000/1000MS (Java/Others)     Memory Limit: 65535/65535KB (Java/Others)

Submit Status

“揽二乔于东南兮，乐朝夕之与共”

一首铜雀台赋，将愤怒与恐惧散播在了孙吴大军之中。

对抗曹军，万事俱备，只欠东风。

现在已经找到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
",found(1,g,h));
        if (l==1) {scanf("%lld",&k);done(1,g,h,k);}
    }
    return 0;
}