xsy2440GDOI2016疯狂动物城

Posted 2020-11-20 dcdcbigbig

感受一下这恐怖的题目长度~~~

其实题意很裸，但是作为GDOI的一道防AK题，自然有他优秀的地方。

简化题意：给出一棵树，要求支持三个操作：

1.修改点值

2.询问点$x$到$y$之间的一些东东

3.回到某个版本之前

可持久化，强制在线。

考虑拆一下询问

如果把$x$到$y$路径上的所有点值放到一个数组$a$里，那么询问式子就是：

$sumlimits_{i=1}^{n}a[i] imessumlimits_{j=1}^{n-i}j$

$=sumlimits_{i=1}^{n}a[i] imesfrac{(n-i+1) imes(n-i)}{2}$

$=sumlimits_{i=1}^{n}a[i] imes i^{2}-a[i] imes(2n+1)+a[i] imes n imes(n+1)$

发现$a[i]$，$a[i] imes i$，$a[i] imes i^2$都是常系数，于是就可以快乐维护了~

外面的很容易想到，树链剖分+可持久化线段树，单点修改即可。

可以感觉到逐渐膨胀的代码量>_<

注意到有左右方向的区别，要写两遍

代码量起飞了>_<

5000B刷新最长代码记录

注意：记得取模，随时爆int；

 

  1 #include<iostream>
  2 #include<cstring>
  3 #include<cstdio>
  4 #include<cmath>
  5 #define mod 20160501
  6 using namespace std;
  7 typedef long long ll;
  8 struct edge{
  9     int v,next;
 10 }a[500001];
 11 struct data{
 12     ll s,a1,a2,s1,s2,ls,rs,lz,vv;
 13 }t[5000001];
 14 ll n,m,u,v,w,ord,tim=0,tot=0,tote=0,ans=0,nrt=0,nowrt=0,last=0,ans1[2],ans2[2],head[1000001],rt[1000001],num[1000001],nmd[1000001],dep[1000001],top[1000001],siz[1000001],dfn[1000001],son[1000001],fa[1000001];
 15 void add(int u,int v){
 16     a[++tote].v=v;
 17     a[tote].next=head[u];
 18     head[u]=tote;
 19 }
 20 void dfs1(int u,int f,int dpt){
 21     fa[u]=f;
 22     dep[u]=dpt;
 23     siz[u]=1;
 24     for(int tmp=head[u];tmp!=-1;tmp=a[tmp].next){
 25         int v=a[tmp].v;
 26         if(v==f)continue;
 27         dfs1(v,u,dpt+1);
 28         siz[u]+=siz[v];
 29         if(son[u]==-1||siz[v]>siz[son[u]])son[u]=v;
 30     }
 31 }
 32 void dfs2(int u,int nowtp){
 33     top[u]=nowtp;
 34     dfn[u]=++tim;
 35     nmd[tim]=u;
 36     if(son[u]==-1)return;
 37     dfs2(son[u],nowtp);
 38     for(int tmp=head[u];tmp!=-1;tmp=a[tmp].next){
 39         int v=a[tmp].v;
 40         if(v==son[u]||v==fa[u])continue;
 41         dfs2(v,v);
 42     }
 43 }
 44 void pushup(int u){
 45     t[u].s=(t[t[u].ls].s+t[t[u].rs].s)%mod;
 46     t[u].a1=(t[t[u].ls].a1+t[t[u].rs].a1)%mod;
 47     t[u].a2=(t[t[u].ls].a2+t[t[u].rs].a2)%mod;
 48     t[u].s1=(t[t[u].ls].s1+t[t[u].rs].s1)%mod;
 49     t[u].s2=(t[t[u].ls].s2+t[t[u].rs].s2)%mod;
 50 }
 51 void build(int l,int r,int u){
 52     if(l==r){
 53         t[u].s=num[nmd[l]];
 54         t[u].a1=dep[nmd[l]];
 55         t[u].a2=(t[u].a1*t[u].a1)%mod;        
 56         t[u].s1=(t[u].s*t[u].a1)%mod;
 57         t[u].s2=(t[u].s*t[u].a2)%mod;
 58         return;
 59     }
 60     int mid=(l+r)/2;
 61     build(l,mid,t[u].ls=++tot);
 62     build(mid+1,r,t[u].rs=++tot);
 63     pushup(u);
 64 }
 65 void updata(int l,int r,int u,int L,int R,int lst,ll v){
 66     if(u==tot)t[u]=t[lst];
 67     if(l==L&&r==R){
 68         t[u].ls=t[lst].ls;
 69         t[u].rs=t[lst].rs;
 70         t[u].lz=(t[u].lz+v)%mod;
 71         t[u].s1=(t[u].s1+t[u].a1*v)%mod;
 72         t[u].s2=(t[u].s2+t[u].a2*v)%mod;
 73         t[u].s=(t[u].s+v*(r-l+1))%mod;
 74         t[u].vv=(t[u].vv+v)%mod;
 75         return;
 76     }
 77     int mid=(l+r)/2;
 78     if(R<=mid){
 79         if(t[u].rs<=last)t[u].rs=t[lst].rs;
 80         if(t[u].ls<=last)t[u].ls=++tot;
 81         updata(l,mid,t[u].ls,L,R,t[lst].ls,v);
 82     }else if(mid<L){
 83         if(t[u].ls<=last)t[u].ls=t[lst].ls;
 84         if(t[u].rs<=last)t[u].rs=++tot;
 85         updata(mid+1,r,t[u].rs,L,R,t[lst].rs,v);
 86     }else{
 87         if(t[u].ls<=last)t[u].ls=++tot;
 88         updata(l,mid,t[u].ls,L,mid,t[lst].ls,v);
 89         if(t[u].rs<=last)t[u].rs=++tot;
 90         updata(mid+1,r,t[u].rs,mid+1,R,t[lst].rs,v);
 91     }
 92     t[u].s1=(t[t[u].ls].s1+t[t[u].rs].s1+t[u].a1*t[u].vv%mod)%mod;
 93     t[u].s2=(t[t[u].ls].s2+t[t[u].rs].s2+t[u].a2*t[u].vv%mod)%mod;
 94     t[u].s=(t[t[u].ls].s+t[t[u].rs].s+t[u].vv*(r-l+1)%mod)%mod;
 95 }
 96 void getans(int u,int b,ll lz,ll v){
 97     ans=(ans+t[u].s2+lz*t[u].a2%mod)%mod;
 98     ans1[b]=(ans1[b]+t[u].s1+t[u].a1*lz%mod)%mod;
 99     ans2[b]=(ans2[b]+t[u].s+v*lz%mod)%mod;
100 }
101 void query(int l,int r,int u,int L,int R,int b,ll lz){
102     if(!u)return;
103     lz+=t[u].vv;
104     if(l==L&&r==R){
105         lz-=t[u].lz;
106         getans(u,b,lz,r-l+1);
107         return;
108     }
109     int mid=(l+r)/2;
110     if(R<=mid)query(l,mid,t[u].ls,L,R,b,lz);
111     else if(mid<L)query(mid+1,r,t[u].rs,L,R,b,lz);
112     else{
113         query(l,mid,t[u].ls,L,mid,b,lz);
114         query(mid+1,r,t[u].rs,mid+1,R,b,lz);
115     }
116 }
117 void work(int t,int t1,int u,int v,ll w){
118     while(top[u]!=top[v]){
119         if(dep[top[u]]<dep[top[v]])swap(u,v);
120         updata(1,n,rt[t],dfn[top[u]],dfn[u],rt[t1],w);
121         u=fa[top[u]];
122     }
123     if(dep[u]>dep[v])swap(u,v);
124     updata(1,n,rt[t],dfn[u],dfn[v],rt[t1],w);
125 }
126 int work1(int t,int u,int v){
127     while(top[u]!=top[v]){
128         if(dep[top[u]]>dep[top[v]]){
129             query(1,n,rt[t],dfn[top[u]],dfn[u],0,0);
130             u=fa[top[u]];
131         }else{
132             query(1,n,rt[t],dfn[top[v]],dfn[v],1,0);
133             v=fa[top[v]];
134         }
135     }
136     if(dep[u]>dep[v]){
137         query(1,n,rt[t],dfn[v],dfn[u],0,0);
138         return v;
139     }else{
140         query(1,n,rt[t],dfn[u],dfn[v],1,0);
141         return u;
142     }
143 }
144 int main(){
145     memset(son,-1,sizeof(son));
146     memset(head,-1,sizeof(head));
147     scanf("%lld%lld",&n,&m);
148     for(int i=1;i<n;i++){
149         scanf("%lld%lld",&u,&v);
150         add(u,v);
151         add(v,u);
152     }
153     for(int i=1;i<=n;i++)scanf("%lld",&num[i]);
154     dfs1(1,0,1);
155     dfs2(1,1);
156     rt[0]=tot=1;
157     build(1,n,rt[0]);
158     for(int i=1;i<=m;i++){
159         scanf("%lld",&ord);
160         if(ord==1){
161             scanf("%lld%lld%lld",&u,&v,&w);
162             u^=ans;
163             v^=ans;
164             last=tot;
165             rt[++nrt]=++tot;
166             work(nrt,nowrt,u,v,w);
167             nowrt=nrt;
168         }else if(ord==2){
169             scanf("%lld%lld",&u,&v);
170             u^=ans;
171             v^=ans;
172             ans=ans1[0]=ans1[1]=ans2[0]=ans2[1]=0;
173             int t=dep[v]-dep[work1(nowrt,u,v)]*2;
174             ans=(ans+ans1[0]*(t*2+1)%mod+ans2[0]*t%mod*(t+1)%mod)%mod;
175             ans=(ans-ans1[1]*(dep[v]*2+1)%mod+ans2[1]*dep[v]%mod*(dep[v]+1)%mod+mod)%mod;
176             ans=(ans*10080251)%(ll)mod;
177             ans=(ans+mod)%mod;
178             printf("%lld
",ans);
179         }else{
180             scanf("%lld",&u);
181             u^=ans;
182             nowrt=u;
183         }
184     }
185     return 0;
186 }