NC14393 点权和(计数)

题目链接

题意：
$一棵n个点的树，最开始点权为0$
$每次选一个点并且把和他距离为1的点的权值加一$
$统计该次加的这些点的最终权值和，乘上次数的总和$
$mod19260817$

题解：
$n<=1e5,m<=1e6$
$数据较大，不能每次暴力统计$
$所以要用数组维护，分开处理$

$考虑对于每次选点的贡献来源$
$肯定是父结点和子结点会给出相应贡献$
$子结点和自身给的贡献就用son[x]维护$
$每次son[x]加上x点的度数，就是当次对x的贡献$
$son[fa[x]]加上2，该次操作对于父结点的贡献是当前结点和父结点两个$
$父亲的父亲结点的son数组加1，该次操作对父亲的父亲结点的贡献$
$就是该次操作父亲结点权值加一，会对父亲的父亲结点产生贡献$
$这样就考虑完了一个结点来自自身和子结点的贡献了$

$然后考虑父节点对选择点的贡献$
$用father[x]来维护$
$对于每次对x操作，x和fa[x]都会增加权值1并对他们的子结点产生贡献$
$所以fa[x]和x的father数组都要加1$
$但发现这样统计的父结点的贡献是不够的$
$因为还有父结点自身操作会增加子结点的权值$
$还有父亲的父亲结点操作增加父亲结点的值没有统计上$
$所以用cnt[x]维护x结点的操作数$
$结果就是$
$father[fa[x]]+son[x]+cnt[fa[x]]+cnt[fa[fa[x]]]$

AC代码

/*
    Author : zzugzx
    Lang : C++
    Blog : blog.csdn.net/qq_43756519
*/
#include<bits/stdc++.h>
using namespace std;
#define fi first
#define se second
#define pb push_back
#define mp make_pair
#define all(x) (x).begin(), (x).end()
#define endl '\n'
#define SZ(x) (int)x.size()
typedef long long ll;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
const int mod = 19260817;
//const int mod = 998244353;
const double eps = 1e-10;
const double pi = acos(-1.0);
const int maxn = 1e6+10;
const ll inf = 0x3f3f3f3f;
const int dir[][2]={{0, 1}, {1, 0}, {0, -1}, {-1, 0}, {1, 1}, {1, -1}, {-1, 1}, {-1, -1}};

ll fa[maxn], d[maxn], son[maxn], father[maxn], cnt[maxn];
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);cout.tie(0);
//  freopen("in.txt", "r", stdin);
//  freopen("out.txt", "w", stdout);
    int n, m;
    cin >> n >> m;
    for (int i = 2; i <= n; i++){
        int x;
        cin >> x;
        fa[i] = x;
        d[x]++; d[i]++;
    }
    ll ans = 0;
    for (int i = 1, x; i <= m; i++) {
        cin >> x;
        ++cnt[x];
        son[x] = (son[x] + d[x]) % mod;
        son[fa[x]] = (son[fa[x]] + 2) % mod;
        son[fa[fa[x]]] = (son[fa[fa[x]]] + 1) % mod;
        father[x] = (father[x] + 1) % mod;
        father[fa[x]] = (father[fa[x]] + 1) % mod;
        ans = (ans + i * (son[x] + father[fa[x]] + cnt[fa[x]] + cnt[fa[fa[x]]])) % mod;
    }
    cout << ans;
    return 0;
}