首页 > 代码库 > AC日记——[WC2013]糖果公园 cogs 1817
AC日记——[WC2013]糖果公园 cogs 1817
[WC2013]糖果公园
思路:
带修改树上莫队(模板);
来,上代码:
#include <cmath>#include <cstdio>#include <cstring>#include <iostream>#include <algorithm>using namespace std;#define maxn 100005#define ll long longstruct QueryType { ll u,v,t,id;};struct QueryType qu[maxn];struct ChangeType { ll to,x,h;};struct ChangeType cha[maxn];ll n,m,q,ti[maxn],totq,totc,vi[maxn],wi[maxn];ll siz,deep[maxn],f[maxn],top[maxn],bel[maxn],size[maxn];ll E[maxn<<1],V[maxn<<1],head[maxn],cnt,dis[maxn],ans[maxn];bool if_[maxn];inline void in(ll &now){ char Cget=getchar();now=0; while(Cget>‘9‘||Cget<‘0‘) Cget=getchar(); while(Cget>=‘0‘&&Cget<=‘9‘) { now=now*10+Cget-‘0‘; Cget=getchar(); }}void pre(ll now,ll fa){ deep[now]=deep[fa]+1,size[now]=1; bel[now]=((++cnt)+1)/siz,f[now]=fa; for(ll i=head[now];i;i=E[i]) { if(V[i]==fa) continue; pre(V[i],now),size[now]+=size[V[i]]; }}void dfs(ll now,ll chain){ ll pos=0;top[now]=chain; for(ll i=head[now];i;i=E[i]) { if(V[i]==f[now]) continue; if(size[V[i]]>size[pos]) pos=V[i]; } if(pos==0) return ; dfs(pos,chain); for(ll i=head[now];i;i=E[i]) { if(V[i]==pos||V[i]==f[now]) continue; dfs(V[i],V[i]); }}ll solve_lca(ll x,ll y){ while(top[x]!=top[y]) { if(deep[top[x]]<deep[top[y]]) swap(x,y); x=f[top[x]]; } if(deep[x]>deep[y]) swap(x,y); return x;}bool cmp(QueryType aa,QueryType bb){ if(bel[aa.u]==bel[bb.u]) { if(bel[aa.v]==bel[bb.v]) return aa.t<bb.t; else return bel[aa.v]<bel[bb.v]; } else return bel[aa.u]<bel[bb.u];}inline void change(ll x){ if(if_[cha[x].to]) { cnt-=wi[ti[dis[cha[x].to]]]*vi[dis[cha[x].to]]; ti[dis[cha[x].to]]--; } cha[x].h=dis[cha[x].to]; dis[cha[x].to]=cha[x].x; if(if_[cha[x].to]) { ti[cha[x].x]++; cnt+=wi[ti[cha[x].x]]*vi[cha[x].x]; }}inline void unchange(ll x){ if(if_[cha[x].to]) { cnt-=wi[ti[cha[x].x]]*vi[cha[x].x]; ti[cha[x].x]--; } dis[cha[x].to]=cha[x].h; if(if_[cha[x].to]) { ti[cha[x].h]++; cnt+=wi[ti[cha[x].h]]*vi[cha[x].h]; }}inline void updata(ll x){ if(if_[x]) { cnt-=wi[ti[dis[x]]]*vi[dis[x]]; ti[dis[x]]--; } else { ti[dis[x]]++; cnt+=wi[ti[dis[x]]]*vi[dis[x]]; } if_[x]=!if_[x];}inline void out(ll x){ if(x>9) out(x/10); putchar(x%10+48);}int main(){ freopen("park.in","r",stdin); freopen("park.out","w",stdout); in(n),in(m),in(q);ll u,v;siz=sqrt(n); for(ll i=1;i<=m;i++) in(vi[i]); for(ll i=1;i<=n;i++) in(wi[i]); for(ll i=1;i<n;i++) { in(u),in(v); E[++cnt]=head[u],V[cnt]=v,head[u]=cnt; E[++cnt]=head[v],V[cnt]=u,head[v]=cnt; } for(ll i=1;i<=n;i++) in(dis[i]); cnt=0,pre(1,0),dfs(1,1);ll ty; for(ll i=1;i<=q;i++) { in(ty); if(ty) { in(qu[++totq].u),in(qu[totq].v),qu[totq].t=totc,qu[totq].id=totq; if(bel[qu[totq].u]>bel[qu[totq].v]) swap(qu[totq].u,qu[totq].v); } else in(cha[++totc].to),in(cha[totc].x); } sort(qu+1,qu+totq+1,cmp),u=1,v=1,cnt=0;ll t=0; for(ll no=1;no<=totq;no++) { while(t<qu[no].t) change(++t); while(t>qu[no].t) unchange(t--); ll lca=solve_lca(u,qu[no].u); while(u!=lca) updata(u),u=f[u];u=qu[no].u; while(u!=lca) updata(u),u=f[u];u=qu[no].u; lca=solve_lca(v,qu[no].v); while(v!=lca) updata(v),v=f[v];v=qu[no].v; while(v!=lca) updata(v),v=f[v];v=qu[no].v; lca=solve_lca(u,v); updata(lca),ans[qu[no].id]=cnt,updata(lca); } for(ll i=1;i<=totq;i++) out(ans[i]),putchar(‘\n‘); fclose(stdin),fclose(stdout); return 0;}
AC日记——[WC2013]糖果公园 cogs 1817
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。