首页 > 代码库 > 2014 UESTC Training for Data Structures C - 东风不与周郎便

2014 UESTC Training for Data Structures C - 东风不与周郎便

C - 东风不与周郎便

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

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

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

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

现在已经找到n个风眼,这些风眼的东风有强有弱,诸葛亮说他每次祈风都能够将一段风眼的东风增强,但需人去帮他布阵。同时他需要时刻掌控风眼的状况,以确定下一步的计划,所以还需要知道一段风眼的强度之和。

借东风,此乃逆天之术,施术者会折阳寿不说,布阵者更是会受十倍之伤。

“何人能当此任?”

“在下愿往,大都督。”

“你是?”

“在下一无名小卒,来自跳蚤街。”

Input

第一行两个整数nm,表示有n个风眼,诸葛亮一共祈风或询问了m次。

第二行n个整数,第i个数ai表示第i个风眼已有东风的强度。

接下来m行,每行开始先读入一个整数si,指明这是一次询问还是一次祈风。

si=0,表明这是一次询问,然后读入两个整数li,ri,表示询问[liri]区间中风眼的东风强度之和。

si=1,表明这是一次祈风,然后读入三个整数li,ri,wi,表示把[liri]区间中每个风眼的东风强度提升wi

1n,m100000,0ai10000,0wi10000,1lirin

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;
}