セグメントツリー(セグメント更新、区間加算lazy操作)

3500 ワード

hdu1556
Color the ball
Time Limit: 9000/3000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 9143    Accepted Submission(s): 4677
Problem Description
N個の風船が一列に並んでいて、左から右に1,2,3.....N.が2個の整数a b(a<=b)を与えるたびに、leleは彼の「小飛鳩」ブランドの電気自動車に乗って風船aから風船bまで順番に各風船に色を塗った.しかしN回以降leleはI個目の風船が何回か色を塗ったことを忘れてしまったので、風船ごとに何回塗られたか算出してもらえますか?
Input
各試験例の第1の挙動は整数N(N<=100000)であり、次のN行は、各行に2つの整数a b(1<=a<=b<=N)を含む.
N=0の場合、入力は終了します.
Output
各試験例は、N個の整数を含む1行を出力し、I番目の数は、I番目の気球が塗りつぶされた回数を表す.
Sample Input

  
    
3 1 1 2 2 3 3 3 1 1 1 2 1 3 0

Sample Output

  
    
1 1 1 3 2 1
<pre name="code" class="cpp">#include"stdio.h"
#include"string.h"
#include"iostream"
#include"map"
#include"string"
#include"queue"
#include"stdlib.h"
#include"math.h"
#define M 100009
#define eps 1e-8
#define inf 1000000000
#define mod 1000000000
#define INF 1000000000
using namespace std;
struct st
{
    int l,r,sum,add,mark;
}a[M*4];
int ans;
void make(int l,int r,int k)
{
    a[k].l=l;
    a[k].r=r;
    a[k].mark=0;
    a[k].add=0;
    if(l==r)
    {
        a[k].sum=0;
        return ;
    }
    int mid=(l+r)/2;
    make(l,mid,k*2);
    make(mid+1,r,k*2+1);
    a[k].sum=a[k*2].sum+a[k*2+1].sum;
}
void updata(int l,int r,int p,int k)
{
    if(a[k].l==l&&a[k].r==r)
    {
        a[k].sum+=(r-l+1);
        a[k].add+=p;//             ,       ,     
        a[k].mark=1;
        return;
    }
    if(a[k].mark)
    {
        a[k*2].sum+=(a[k*2].r-a[k*2].l+1)*a[k].add;
        a[k*2].add+=a[k].add;
        a[k*2].mark=1;
        a[k*2+1].sum+=(a[k*2+1].r-a[k*2+1].l+1)*a[k].add;
        a[k*2+1].add+=a[k].add;
        a[k*2+1].mark=1;
        a[k].mark=0;
        a[k].add=0;//        ,         0,
    }
    int mid=(a[k].l+a[k].r)/2;
    if(r<=mid)
        updata(l,r,p,k*2);
    else if(l>mid)
        updata(l,r,p,k*2+1);
    else
    {
        updata(l,mid,p,k*2);
        updata(mid+1,r,p,k*2+1);
    }
    a[k].sum=a[k*2].sum+a[k*2+1].sum;
}
void query(int p,int k)
{
    if(a[k].l==p&&a[k].r==p)
    {
        ans+=a[k].sum;
        return;
    }
    if(a[k].mark)
    {
        a[k*2].sum+=(a[k*2].r-a[k*2].l+1)*a[k].add;
        a[k*2].add+=a[k].add;
        a[k*2].mark=1;
        a[k*2+1].sum+=(a[k*2+1].r-a[k*2+1].l+1)*a[k].add;
        a[k*2+1].add+=a[k].add;
        a[k*2+1].mark=1;
        a[k].mark=0;
        a[k].add=0;
    }
    int mid=(a[k].l+a[k].r)/2;
    if(p<=mid)
        query(p,k*2);
    else
        query(p,k*2+1);
    a[k].sum=a[k*2].sum+a[k*2+1].sum;
}
int main()
{
    int n,i;
    while(scanf("%d",&n),n)
    {
        make(1,n,1);
        int x,y;
        for(i=1;i<=n;i++)
        {
            scanf("%d%d",&x,&y);
            updata(x,y,1,1);
        }
        for(i=1;i<=n;i++)
        {
            ans=0;
            query(i,1);
            if(i==1)
            printf("%d",ans);
            else
            printf(" %d",ans);
        }
        printf("
"); } return 0; }