2631: tree
Time Limit: 30 Sec Memory Limit: 128 MB
Submit: 1486 Solved: 515
[Submit][Status]
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
Source
题解：
在pushdown上加一些东西就可以。坑爹的题目居然还要用unsigned int
我惊呆了，long long 居然T了，差这么多。。。
代码：
1 #include<cstdio>
2 #include<cstdlib>
3 #include<cmath>
4 #include<cstring>
5 #include<algorithm>
6 #include<iostream>
7 #include<vector>
8 #include<map>
9 #include<set>
10 #include<queue>
11 #include<string>
12 #define inf 1000000000
13 #define maxn 100000+100
14 #define maxm 500+100
15 #define eps 1e-10
16 #define ll long long
17 #define pa pair<int,int>
18 #define mod 51061
19 using namespace std;
20 inline int read()
21 {
22 int x=0,f=1;char ch=getchar();
23 while(ch<‘0‘||ch>‘9‘){if(ch==‘-‘)f=-1;ch=getchar();}
24 while(ch>=‘0‘&&ch<=‘9‘){x=10*x+ch-‘0‘;ch=getchar();}
25 return x*f;
26 }
27 struct edge{int go,next;}e[2*maxn];
28 unsigned int n,m,tot,head[maxn],fa[maxn],c[maxn][2],s[maxn];
29 unsigned int sta[maxn],add[maxn],mul[maxn],sum[maxn],v[maxn];
30 bool rev[maxn];
31 inline void insert(int x,int y)
32 {
33 e[++tot].go=y;e[tot].next=head[x];head[x]=tot;
34 e[++tot].go=x;e[tot].next=head[y];head[y]=tot;
35 }
36 inline bool isroot(int x)
37 {
38 return c[fa[x]][0]!=x&&c[fa[x]][1]!=x;
39 }
40 inline void pushup(int x)
41 {
42 s[x]=s[c[x][0]]+s[c[x][1]]+1;
43 sum[x]=(sum[c[x][0]]+sum[c[x][1]]+v[x])%mod;
44 }
45 inline void rotate(int x)
46 {
47 int y=fa[x],z=fa[y],l=(c[y][1]==x),r=l^1;
48 if(!isroot(y))c[z][c[z][1]==y]=x;
49 fa[x]=z;fa[y]=x;fa[c[x][r]]=y;
50 c[y][l]=c[x][r];c[x][r]=y;
51 pushup(y);pushup(x);
52 }
53 inline void update(int x,int y,int z)
54 {
55 if(!x)return;
56 add[x]=(add[x]*z+y)%mod;mul[x]=(mul[x]*z)%mod;
57 sum[x]=(sum[x]*z+s[x]*y)%mod;v[x]=(v[x]*z+y)%mod;
58 }
59 inline void pushdown(int x)
60 {
61 int l=c[x][0],r=c[x][1];
62 if(rev[x])
63 {
64 rev[x]^=1;rev[l]^=1;rev[r]^=1;
65 swap(c[x][0],c[x][1]);
66 }
67 update(l,add[x],mul[x]);update(r,add[x],mul[x]);
68 add[x]=0;mul[x]=1;
69 }
70 inline void splay(int x)
71 {
72 int top=0;sta[++top]=x;
73 for(int y=x;!isroot(y);y=fa[y])sta[++top]=fa[y];
74 for(;top;)pushdown(sta[top--]);
75 while(!isroot(x))
76 {
77 int y=fa[x],z=fa[y];
78 if(!isroot(y))
79 {
80 if(c[z][0]==y^c[y][0]==x)rotate(x);else rotate(y);
81 }
82 rotate(x);
83 }
84
85 }
86 inline void access(int x)
87 {
88 for(int y=0;x;x=fa[x])
89 {
90 splay(x);c[x][1]=y;pushup(x);y=x;
91 }
92 }
93 inline void makeroot(int x)
94 {
95 access(x);splay(x);rev[x]^=1;
96 }
97 inline int find(int x)
98 {
99 access(x);splay(x);
100 while(c[x][0])x=c[x][0];
101 return x;
102 }
103 inline void link(int x,int y)
104 {
105 makeroot(x);fa[x]=y;splay(x);
106 }
107 inline void cut(int x,int y)
108 {
109 makeroot(x);access(y);splay(y);c[y][0]=fa[x]=0;
110 }
111 inline void split(int x,int y)
112 {
113 makeroot(x);access(y);splay(y);
114 }
115 int main()
116 {
117 freopen("input.txt","r",stdin);
118 freopen("output.txt","w",stdout);
119 n=read();m=read();
120 int x,y,z,top=0;char ch[10];
121 for(int i=1;i<n;i++)x=read(),y=read(),insert(x,y);
122 for(int i=1;i<=n;i++)s[i]=sum[i]=v[i]=mul[i]=1,add[i]=0;
123 sta[++top]=1;
124 for(int i=1;i<=top;i++)
125 for(int k=sta[i],j=head[k];j;j=e[j].next)
126 if(e[j].go!=fa[k])
127 {
128 fa[y=e[j].go]=k;sta[++top]=y;
129 }
130 while(m--)
131 {
132 scanf("%s",ch);
133 switch(ch[0])
134 {
135 case ‘-‘:{cut(read(),read());link(read(),read());}break;
136 case ‘/‘:{x=read();y=read();split(x,y);printf("%d\n",sum[y]);}break;
137 case ‘+‘:{x=read();y=read();z=read();split(x,y);update(y,z,1);}break;
138 case ‘*‘:{x=read();y=read();z=read();split(x,y);update(y,0,z);}break;
139 }
140 }
141 return 0;
142 }
View Cod
时间： 2024-12-11 23:23:27