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

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;

}

2014 UESTC Training for Data Structures C - 东风不与周郎便,码迷,mamicode.com