HDU 1166 敌兵布阵

以前用树状数组AC的。

今天用线段树重写了一下。

328ms。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int ql,qr,n;
int a[50001],sum[500001];
int query(int l,int r,int o)
{
    int m=(l+r)>>1,ans=0;
    if(ql<=l&&qr>=r)return sum[o];
    if(ql<=m)ans+=query(l,m,o*2);
    if(qr>m)ans+=query(m+1,r,o*2+1);
    return ans;
}
int update(int l,int r,int o,int index,int ans)
{
    int m=(l+r)>>1;
    if(l<r)
    {
        int x=sum[o*2],y=sum[o*2+1];
        if(index<=m)
        x=update(l,m,o*2,index,ans);
        else
        y=update(m+1,r,o*2+1,index,ans);
        sum[o]=x+y;
    }
    else sum[o]+=ans;
    return sum[o];
}
int build(int l,int r,int o)
{
    int m=(l+r)>>1;
    if(l<r)
    {
        sum[o]=build(l,m,o*2)+build(m+1,r,o*2+1);
    }
    else
    sum[o]=a[m];
    return sum[o];
}
int main()
{
    int t;
    scanf("%d",&t);
    for(int j=1;j<=t;j++)
    {
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
        build(1,n,1);
        char str[10];
        int x,y;
        printf("Case %d:\n",j);
        while(1)
        {
            scanf("%s",str);
            if(!strcmp(str,"Query"))
            {
                scanf("%d%d",&ql,&qr);
                printf("%d\n",query(1,n,1));
            }
            else if(!strcmp(str,"Add"))
            {
                scanf("%d%d",&x,&y);
                update(1,n,1,x,y);
                a[x]+=y;
            }
            else if(!strcmp(str,"Sub"))
            {
                scanf("%d%d",&x,&y);
                update(1,n,1,x,-y);
                a[x]-=y;
            }
            else if(!strcmp(str,"End"))
            break;
        }
    }
}

以前写的树状数组

256ms。

#include<stdio.h>
#include<string.h>
#include<math.h>
int a[50001],n;
int c[50001],m;
int shift()
{
    int i,j,k,sum;
    for(i=1;i<=n;i++)
    {
        k=i&(-i);sum=0;
        for(j=i;j>=i-k+1;j--)
        sum+=a[j];
        c[i]=sum;
    }
}
int query(int x,int y)
{
    int sumx=0,sumy=0;
    x--;
    while(x>0)
    {
        sumx+=c[x];
        x-=x&(-x);
    }
    while(y>0)
    {
        sumy+=c[y];
        y-=y&(-y);
    }
    printf("%d\n",sumy-sumx);
}
int add(int x,int y)
{
    while(x<=n)
    {
        c[x]+=y;
        x+=x&(-x);
    }
}
int main()
{
    int i,x,t,j,y;char str[10];
    scanf("%d",&t);
    for(j=1;j<=t;j++)
    {
        scanf("%d",&n);
        for(i=1;i<=n;i++)
        scanf("%d",&a[i]);
        shift();
        printf("Case %d:\n",j);
        while(scanf("%s",str),strcmp(str,"End")!=0)
        {
            scanf("%d%d",&x,&y);
            if(!strcmp(str,"Query"))query(x,y);
            else if(!strcmp(str,"Add"))add(x,y);
            else if(!strcmp(str,"Sub"))add(x,-y);
        }
    }
}

今天用ZKW线段树写了一次。

ORZ一下 ZKW。281ms。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int P;
int a[500001];
int query(int l,int r)
{
    int ans=0;
    for(l=l+P-2,r=r+P;l^r^1;l>>=1,r>>=1)
    {
        if(~l&1)ans+=a[l^1];
        if(r&1)ans+=a[r^1];
    }
    return ans;
}
void update(int n,int x)
{
    a[n+=P]+=x;
    for(n>>=1;n;n>>=1)
    {
        a[n]=a[n*2]+a[n*2+1];
    }
}
void build(int l,int r)
{
    int i,j;
    for(l>>=1;l>0;l>>=1)
    {
        i=l,r>>=1;
        for(j=i;j<=r;j++)
        a[j]=a[j*2]+a[j*2+1];
    }
}
int main()
{
    int t,n;
    scanf("%d",&t);
    for(int k=1;k<=t;k++)
    {
        scanf("%d",&n);
        for(P=1;P<=n+1;P*=2);
        for(int i=P;i<=n+P;i++)
        scanf("%d",&a[i]);
        build(P,P+n);
        char str[10];
        int l,r;
        printf("Case %d:\n",k);
        while(1)
        {
            scanf("%s",str);
            if(!strcmp(str,"End"))break;
            else if(!strcmp(str,"Query"))
            {
                scanf("%d%d",&l,&r);
                printf("%d\n",query(l,r));
            }
            else if(!strcmp(str,"Add"))
            {
                scanf("%d%d",&l,&r);
                update(l-1,r);
            }
            else if(!strcmp(str,"Sub"))
            {
                scanf("%d%d",&l,&r);
                update(l-1,-r);
            }
        }
    }
}