BZOJ2631tree

10 篇文章 0 订阅

2631: tree
Time Limit: 30 Sec Memory Limit: 128 MB
Submit: 3070 Solved: 1033
Description
 一棵n个点的树,每个点的初始权值为1。对于这棵树有q个操作,每个操作为以下四种操作之一:
+ u v c:将u到v的路径上的点的权值都加上自然数c;
- u1 v1 u2 v2:将树中原有的边(u1,v1)删除,加入一条新边(u2,v2),保证操作完之后仍然是一棵树;
* u v c:将u到v的路径上的点的权值都乘上自然数c;
/ u v:询问u到v的路径上的点的权值和,求出答案对于51061的余数。
Input
第一行两个整数n,q
接下来n-1行每行两个正整数u,v,描述这棵树
接下来q行,每行描述一个操作
Output
对于每个/对应的答案输出一行
Sample Input
3 2
1 2
2 3
* 1 3 4
/ 1 1
Sample Output
4
HINT
数据规模和约定
10%的数据保证,1<=n,q<=2000
另外15%的数据保证,1<=n,q<=5*10^4,没有-操作,并且初始树为一条链
另外35%的数据保证,1<=n,q<=5*10^4,没有-操作
100%的数据保证,1<=n,q<=10^5,0<=c<=10^4
LCT水过。。
虽说是1A,但是从中(Claris学长)学到了不少东西。。
rever是把x设置为根
access是打通x到根的路径
WA点:
①修改的时候忽略了点本身的权值。。
②对于加法标记赋值为了1(是不是很二。。)
③access模板一定要打对。。
练习了一下链修改、链查询、加边、删边。。
感谢Claris神犇。。
附上本蒟蒻的代码:

#include<cstdio>
#include<iostream>
using namespace std;
#define mod 51061
int n,q,father[100001],c[100001][2],st[100001];
bool rev[100001];
unsigned int sum[100001],add[100001],mul[100001],size[100001],val[100001];

int read()
{
    int w=0,f=1; char ch=getchar();
    while (ch<'0' || ch>'9')
      {
        if (ch=='-') f=-1;
        ch=getchar();
      }
    while (ch>='0' && ch<='9')
      w=w*10+ch-'0',ch=getchar();
    return w*f;
}

bool isroot(int x)
{
    return c[father[x]][0]!=x && c[father[x]][1]!=x;
}

void cal(int x,int m,int a)
{
    if (!x) return;
    val[x]=(val[x]*m+a)%mod;
    sum[x]=(sum[x]*m+a*size[x])%mod;
    add[x]=(add[x]*m+a)%mod;
    mul[x]=(mul[x]*m)%mod;
}

void update(int x)
{
    int l=c[x][0],r=c[x][1];
    sum[x]=(sum[l]+sum[r]+val[x])%mod;
    size[x]=(size[l]+size[r]+1)%mod;
}

void pushdown(int k)
{
    int l=c[k][0],r=c[k][1],m=mul[k],a=add[k];
    if (rev[k]) rev[k]^=1,rev[l]^=1,rev[r]^=1,swap(c[k][0],c[k][1]);
    mul[k]=1,add[k]=0;
    if (m!=1 || a!=0) cal(l,m,a),cal(r,m,a);
}

void rotate(int x)
{
    int y=father[x],z=father[y],l,r;
    if (c[y][0]==x) l=0;
    else l=1;
    r=l^1;
    if (!isroot(y))
      if (c[z][0]==y) c[z][0]=x;
      else c[z][1]=x;
    father[x]=z,father[y]=x,father[c[x][r]]=y;
    c[y][l]=c[x][r],c[x][r]=y;
    update(y),update(x);
}

void splay(int x)
{
    int i,top=0,y,z;
    top++,st[top]=x;
    for (i=x;!isroot(i);i=father[i])
      top++,st[top]=father[i];
    for (i=top;i;i--)
      pushdown(st[i]);
    while (!isroot(x))
      {
        y=father[x],z=father[y];
        if (!isroot(y))
          if (c[y][0]==x^c[z][0]==y) rotate(x);
          else rotate(y);
        rotate(x);
      }
}

void access(int x)
{
    int t;
    for (t=0;x;t=x,x=father[x])
      splay(x),c[x][1]=t,update(x);
}

void rever(int x)
{
    access(x),splay(x),rev[x]^=1;
}

void link(int x,int y)
{
    rever(x),father[x]=y;
}

void cut(int x,int y)
{
    rever(x),access(y),splay(y),c[y][0]=father[x]=0;
}

void split(int x,int y)
{
    rever(y),access(x),splay(x);
}

int main()
{
    char s[10];
    int i,x,y,z;
    n=read(),q=read();
    for (i=1;i<=n;i++)
      sum[i]=mul[i]=size[i]=val[i]=1;
    for (i=1;i<=n-1;i++)
      x=read(),y=read(),link(x,y);
    for (i=1;i<=q;i++)
      {
        scanf("%s",&s),x=read(),y=read();
        if (s[0]=='+')
          z=read(),split(x,y),cal(x,1,z);
        if (s[0]=='-')
          cut(x,y),x=read(),y=read(),link(x,y);
        if (s[0]=='*')
          z=read(),split(x,y),cal(x,z,0);
        if (s[0]=='/')
          split(x,y),printf("%d\n",sum[x]);
      }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值