fzu 2171 线段树 lazy标记

时间：2014-05-07 18:41:12 阅读：440 评论：0 收藏：0 [点我收藏+]

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

Problem 2171 防守阵地 II

Accept: 73 Submit: 256
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 1 2 3

Sample Output

6 3 5

#include<iostream>
#include<stdio.h>
#include<string.h>
#define maxx 100002
using namespace std;

struct node{int l,r,color,sum;}tree[maxx*4];

void build(int left,int right,int k)
{
    tree[k].l=left;
    tree[k].r=right;
    tree[k].color=0;
    if(left==right)
    {
        scanf("%d",&tree[k].sum);
        return;
    }
    int m;
    m=(left+right)/2;
    build(left,m,k*2);
    build(m+1,right,k*2+1);
    tree[k].sum=tree[k*2].sum+tree[k*2+1].sum;
}

void pushdown(int k)
{
    tree[k*2].sum-=(tree[k*2].r-tree[k*2].l+1)*tree[k].color;
    tree[k*2+1].sum-=(tree[k*2+1].r-tree[k*2+1].l+1)*tree[k].color;
    tree[k*2].color+=tree[k].color;
    tree[k*2+1].color+=tree[k].color;
}

int sum(int left,int right,int k)
{
    if(tree[k].l==left&&tree[k].r==right)
        return tree[k].sum;
    if(tree[k].color)
    {
        pushdown(k);
        tree[k].color = 0;
    }
    int m;
    m=(tree[k].l+tree[k].r)/2;
    if(right<=m)
        return sum(left,right,k*2);
    if(left>=m+1)
        return sum(left,right,k*2+1);
    return sum(left,m,k*2)+sum(m+1,right,k*2+1);
}

void update(int left,int right,int k)
{
    if(tree[k].l==left&&tree[k].r==right)
    {
        tree[k].sum-=tree[k].r-tree[k].l+1;
        tree[k].color++;
        return ;
    }
    int m;
    m=(tree[k].l+tree[k].r)/2;
    if(right<=m)
        update(left,right,k*2);
    else if(left>=m+1)
        update(left,right,k*2+1);
    else
    {
        update(left,m,k*2);
        update(m+1,right,k*2+1);

    }tree[k].sum=tree[k*2].sum+tree[k*2+1].sum;
}



int main()
{
    int i,j,n,m,t,q,a,b;
    while(~scanf("%d%d%d",&n,&m,&q))
    {
        build(1,n,1);
        while(q--)
        {
            //printf("1");
            scanf("%d",&a); //printf(" er ");
            b=a+m-1;
            printf("%d\n",sum(a,b,1));//printf("sdfd");
            update(a,b,1);//printf("okok");
        }
    }
}