nowcode练习赛6（B）:点权和

牛客网练习赛6（B）：
链接： https://www.nowcoder.com/acm/contest/26/B
思路：
因为m的大小为1e7，那么怎么做到O(1)的维护节点的儿子和父亲的信息；
fa[x]表示x的父亲的编号；
用一个sum[x]数组记录节点x的所有儿子节点的权值；
用一个cnt[x]数组表示当前 x节点出现的次数；
dep[x]表示x节点的儿子的个数，即出度；
arr[x]为x节点的权值；
那么，对于每次的的询问，维护

arr[x]++;
cnt[x]++;
sum[x] += dep[x];
sum[x] %= mod;
arr[fa[x]]++;
sum[fa[x]]++;
sum[fa[fa[x]]]++;

那么每次的ans:
ans儿子：sum[x];
ans自己：arr[x]+cnt[fa[x]]];
ans父亲：arr[fa[x]]+cnt[fa[fa[x]]];
ans = ans儿子+ans自己+ans父亲;
code:

#include<bits/stdc++.h>
using namespace std;
typedef int INT;
const int maxn = 1e5 + 99;
const int mod = 19260817;
int fa[maxn];
int arr[maxn];
int sum[maxn];
int dep[maxn];
int cnt[maxn];
inline bool in(int &ret)
{
    char ch;
    bool sgn = false;//正数
    ret = 0;
    if ((ch = getchar()) == EOF)
        return false;//EOF
    while (ch != '-' && (ch<'0' || ch>'9'))
        ch = getchar();
    if (ch == '-') sgn = true;//负数
    else ret = ch - '0';
    while (ch = getchar(), ch >= '0'&&ch <= '9')
        ret = ret * 10 + ch - '0';
    if (sgn) ret = -ret;
    return true;//当前输入结束
}
INT main()
{
    int n, m;
    scanf("%d%d", &n, &m);
    for (int i = 1; i < n; ++i)
    {
        int x;
        in(x);
        fa[i + 1] = x;
        dep[x]++;
    }
    int Hash = 0;
    int i = 1;
    fa[1] = 0;
    while (m--)
    {
        int ans = 0, x;
        in(x);
        arr[x]++;
        cnt[x]++;
        sum[x] += dep[x];
        sum[x] %= mod;
        arr[fa[x]]++;
        sum[fa[x]]++;
        sum[fa[fa[x]]]++;
        sum[0] = 0; cnt[0] = 0; arr[0] = 0;//这个不要漏了，防止x的父亲为0的情况；
        ans = (arr[x] + arr[fa[x]] + sum[x] + cnt[fa[x]] + cnt[fa[fa[x]]]) % mod;
        Hash = (Hash + (long long)ans*i%mod) % mod;
        i++;
    }
    printf("%d\n", Hash);
    return 0;
}

END