fzu 2171 线段树区间更新

时间：2014-07-16 15:25:12 阅读：212 评论：0 收藏：0 [点我收藏+]

http://acm.fzu.edu.cn/problem.php?pid=2171

Problem 2171 防守阵地 II

Accept: 105 Submit: 415
Time Limit: 3000 mSec Memory Limit : 32768 KB

Problem Description

部队中总共有N个士兵，每个士兵有各自的能力指数Xi，在一次演练中，指挥部确定了M个需要防守的地点，指挥部将选择M个士兵依次进入指定地点进行防守任务，获得的参考指数即为M个士兵的能力之和。随着时间的推移，指挥部将下达Q个指令来替换M个进行防守的士兵们，每个参加完防守任务的士兵由于疲惫等原因能力指数将下降1。现在士兵们排成一排，请你计算出每次进行防守的士兵的参考指数。

Input

输入包含多组数据。

输入第一行有两个整数N,M,Q(1<=N<=100000，1<=M<=1000,1<=Q<=100000),第二行N个整数表示每个士兵对应的能力指数Xi（1<=Xi<=1000）。

接下来Q行，每行一个整数X，表示在原始队列中以X为起始的M个士兵替换之前的士兵进行防守。（1<=X<=N-M+1）

对于30%的数据1<=M,N,Q<=1000。

Output

输出Q行，每行一个整数，为每次指令执行之后进行防守的士兵参考指数。

Sample Input

5 3 3

2 1 3 1 4

Sample Output

========================================================

经典的线段树区间更新，这里只贴个代码，自己尚待理解

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream>

#define Maxx 100010

using namespace std;
int str[Maxx];
long long ans;
struct Node
{
    int left,right,add,maxnode,sum;
};

Node tree[Maxx<<2];

void build(int l,int r,int rt)
{
    tree[rt].left=l;
    tree[rt].right=r;
    tree[rt].add=0;

    if(l == r)
    {
        tree[rt].sum=str[l];
        return ;
    }

    int mid=(l+r)/2;

    build(l,mid,rt*2);
    build(mid+1,r,rt*2+1);
    //return 0;
    tree[rt].sum=tree[rt*2].sum+tree[rt*2+1].sum;
}

void update(int l,int r,int rt,int add)
{
    if(tree[rt].left>r || tree[rt].right<l)
    {
        return ;
    }

    if(tree[rt].left >=l&&tree[rt].right <=r)
    {
        tree[rt].sum+=(tree[rt].right-tree[rt].left+1)*add;
        tree[rt].add+=add;
        return ;
    }

    if(tree[rt].add)
    {
        tree[rt*2].sum+=(tree[rt*2].right-tree[rt*2].left+1)*tree[rt].add;
        tree[rt*2].add+=tree[rt].add;
        tree[rt*2+1].sum+=(tree[rt*2+1].right-tree[rt*2+1].left+1)*tree[rt].add;
        tree[rt*2+1].add+=tree[rt].add;
        tree[rt].add=0;
    }

    update(l,r,rt*2,add);
    update(l,r,rt*2+1,add);
    tree[rt].sum=tree[rt*2].sum+tree[rt*2+1].sum;
}

void query(int l,int r,int rt)
{
    if(tree[rt].left>r || tree[rt].right<l)
    {
        return ;
    }

    if(tree[rt].left>=l && tree[rt].right<=r)
    {
        ans+=tree[rt].sum;
        return ;
    }

    if(tree[rt].add)
    {
        tree[rt*2].sum+=(tree[rt*2].right-tree[rt*2].left+1)*tree[rt].add;
        tree[rt*2].add+=tree[rt].add;
        tree[rt*2+1].sum+=(tree[rt*2+1].right-tree[rt*2+1].left+1)*tree[rt].add;
        tree[rt*2+1].add+=tree[rt].add;
        tree[rt].add=0;
    }
    query(l,r,rt*2);
    query(l,r,rt*2+1);
    tree[rt].sum=tree[rt*2].sum+tree[rt*2+1].sum;
    //return ans;
}

int main()
{
    int n,m,q;
    int x,i,j;
    while(scanf("%d%d%d",&n,&m,&q)!=EOF)
    {
        for(i=1;i<=n;i++)
        {
            scanf("%d",&str[i]);
        }
        build(1,n,1);
        for(i=0;i<q;i++)
        {
            scanf("%d",&x);
            ans=0;
            query(x,x+m-1,1);
            printf("%lld\n",ans);
            update(x,x+m-1,1,-1);
        }
    }
    return 0;
}