2018年湖南省赛主席树题

题意：给你一棵树，要你求出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]);
    }
}