题意:给你一棵树,要你求出1号节点到其他节点的有序对的个数。
思路代码来自:ccsu_cat
思路:树上建可持续化权值线段树。ans是每个节点的答案,a是权值。对于一条链来说,如果前面没有出现过这个值,我的ans【i】=ans【i-1】+ p -1。如果出现一次,我就需要用一个d数组去保存a【i】出现的位置,此时我的ans【i】=ans【i-1】+i-d【a【i】】+1((a【i】,a【i】)对答案的贡献)如果这个数以前出现了二次及以上,我的ans【i】=ans【i】+i-d【a【i】】(因为我(a【i】,a【i】)对答案的贡献已经加上了),很容易想到但是我没有想到的是,d数组可以可以这样写,当,一条链以前没有出现过x时,d【x】=0;第一次出现的时候,我的d就等于x的前一个位置,我的答案就+=i-1。以后再出现的时候,我们的答案就直接等于i-d【x】,这样就可以保证(x,x)加到答案中了并且只加了一次。
分为两种情况,第一种情况是,a【i】没又出现过,那么我的ans【i】=ans【i-1】+ i - 1;d【a【i】】=i-1,第二种情况是:我的a【i】出现过,即我的d【a【i】】!=-1,(为什么是等于-1?因为假设等于0,当我的a【i】的第一个位置是1是i-1=0,我的d【a【i】】又等于0,第一种情况就会执行两次),那么我的ans【i】=ans【i-1】+ i-d【a【i】】。
需要注意的是,由于他给我们的是一棵树,有许多条链,对于每一条链来说,我们我们d【a【i】】保存的是这条链上的信息,在dfs结束后,意味着这条链已经遍历完成,那么,我可以保存d【a【u】】的前一个值,在到达叶子节点值将他赋给d【a【u】】。
附上ac的代码,根据某大佬ac代码补的
#include<bits/stdc++.h> using namespace std; const int maxn=1e5+10; int n,rt[maxn],tr[maxn*20],ls[maxn*20],rs[maxn*20],a[maxn],d[maxn]; long long cnt,ans[maxn]; vector<int>G[maxn]; void up(int &o,int pre,int l,int r,int p) { o=++cnt; ls[o]=ls[pre]; rs[o]=rs[pre]; tr[o]=tr[pre]+1; if(l==r)return ; int mid=(l+r)/2; if(p<=mid)up(ls[o],ls[pre],l,mid,p); else up(rs[o],rs[pre],mid+1,r,p); tr[o]=tr[ls[o]]+tr[rs[o]]; } void dfs(int u,int fa) { ans[u]=ans[fa]; rt[u]=rt[fa]; int tmp=d[a[u]]; if(d[a[u]]==-1) { up(rt[u],rt[fa],1,n,a[u]); d[a[u]]=tr[rt[u]]-1; ans[u]+=d[a[u]]; } else { ans[u]+=tr[rt[u]]-d[a[u]]; d[a[u]]=tr[rt[u]]; } for(int i=0;i<G[u].size();i++) { int v=G[u][i]; if(v==fa)continue; dfs(v,u); } d[a[u]]=tmp; } int main() { while(~scanf("%d",&n)) { cnt=0; for(int i=1;i<n+10;i++)d[i]=-1,G[i].clear(); for(int i=2;i<=n;i++) { int u; scanf("%d",&u); G[u].push_back(i); G[i].push_back(u); } for(int i=1;i<=n;i++)scanf("%d",&a[i]); dfs(1,0); for(int i=2;i<=n;i++)printf("%lld\n",ans[i]); } }