码迷,mamicode.com
首页 > 其他好文 > 详细

reshuffle

时间:2019-11-19 17:09:49      阅读:9      评论:0      收藏:0      [点我收藏+]

标签:build   com   void   can   efi   while   scanf   case   cas   

线段树

pragma comment(linker, "/STACK:1024000000,1024000000") //HDU 5692

include

include

include

include

using namespace std;

define maxn 400010

define INF 1e18

struct re
{
int x,y,rz;
long long z;
};
re f[maxn];

int fi[maxn],ne[maxn],u[maxn],v[maxn];
bool vis[maxn];
int t=0,step=0;
int a[maxn];
long long d[maxn];
int data[maxn];
int l[maxn],r[maxn];

long long max(long long a,long long b)
{
if (a>b) return a;else return b;
}
void edge(int x,int y)
{
t++;
ne[t]=fi[x];
fi[x]=t;
u[t]=x;
v[t]=y;
t++;
ne[t]=fi[y];
fi[y]=t;
u[t]=y;
v[t]=x;
}
void dfs(int now)
{
step++;
a[step]=now;
l[now]=step;
vis[now]=1;
for (int i=fi[now];i!=-1;i=ne[i]) if (!vis[v[i]])
{
int next=v[i];
d[next]+=d[now];
dfs(next);
}
r[now]=step;
return;
}

void down(int k)
{
f[k*2].rz+=f[k].rz;
f[k*2].z+=f[k].rz;

f[k*2+1].rz+=f[k].rz;
f[k*2+1].z+=f[k].rz;    

f[k].rz=0;
return;

}

void up(int k)
{
f[k].z=max(f[k*2].z,f[k*2+1].z);
return;
}

void update(int k,int l,int r,int zd)
{
if ((f[k].x>r)||(f[k].y<l)) return;
if ((f[k].x>=l)&&(f[k].y<=r))
{
f[k].rz+=zd;
f[k].z+=zd;
// int i=k; printf("%d %d %d %d %lf %d\n",l,r,f[i].x,f[i].y,f[i].z,f[i].rz);
return;
}

down(k);
update(k*2,l,r,zd);
update(k*2+1,l,r,zd);       

up(k);

// int i=k; printf("%d %d %d %d %lf %d\n",l,r,f[i].x,f[i].y,f[i].z,f[i].rz);
return;
}

long long inquiry(int k,int l,int r)
{

// printf("%d %d %d %d %lld\n",l,r,f[k].x,f[k].y,f[k].z);

if ((f[k].x>r)||(f[k].y<l)) return -INF;
if ((f[k].x>=l)&&(f[k].y<=r))
{
    return f[k].z;
}           
down(k);
long long aa=inquiry(k*2,l,r);
long long bb=inquiry(k*2+1,l,r);    
return max(aa,bb);

}

void build(int k,int l,int r)
{
f[k].x=l;
f[k].y=r;
if (l==r)
{
f[k].z=d[a[l]];
f[k].rz=0;
return;
}
int mid=(l+r)/2;
build(k2,l,mid);
build(k
2+1,mid+1,r);
up(k);
return;
}

int main()
{
int turn;
cin>>turn;
int kase=0;
while (turn--)
{

    printf("Case #%d:\n",++kase);
            
    t=0,step=0;
    
    int n,m;
    cin>>n>>m;
    for (int i=0;i<=n;i++) fi[i]=-1;
    for (int i=1;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        edge(x,y);
    }
    for (int i=0;i<n;i++)
    {
        scanf("%d",&data[i]);
        d[i]=data[i];
    }
    memset(vis,0,sizeof(vis));
    dfs(0);

// for (int i=0;i<n;i++) printf("%d %d %d %d\n",i,l[i],r[i],d[i]);

    build(1,1,n);

// for (int i=1;i<=15;i++) printf("%d %d %d %lld\n",i,f[i].x,f[i].y,f[i].z);

    for (int i=1;i<=m;i++)
    {
        int qwe;
        scanf("%d",&qwe);
        if (qwe==1)
        {
            int xx;
            scanf("%d",&xx);
            printf("%lld\n",inquiry(1,l[xx],r[xx]));
        }
        else if (qwe==0)
        {
            int xx,yy;
            scanf("%d%d",&xx,&yy);
            int delta=yy-data[xx];
            data[xx]=yy;
            update(1,l[xx],r[xx],delta);
        }
    }

}

return 0;

}

reshuffle

标签:build   com   void   can   efi   while   scanf   case   cas   

原文地址:https://www.cnblogs.com/reshuffle/p/11890518.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有 京ICP备13008772号-2
迷上了代码!