点权和

点权和

https://ac.nowcoder.com/acm/problem/14393

思路:建树->对树进行更新维护->输出

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cctype>
#include <algorithm>
using namespace std;
#define rep(i, s, t) for(int i = (s); i <= (t); i++)
#define dwn(i, s, t) for(int i = (s); i >= (t); i--)

int read() {
    int x = 0, f = 1; char c = getchar();
    while(!isdigit(c)){ if(c == '-') f = -1; c = getchar(); }
    while(isdigit(c)){ x = x * 10 + c - '0'; c = getchar(); }
    return x * f;
}

#define maxn 100010
#define MOD 19260817
#define LL long long

int n, q, fa[maxn], m, head[maxn], nxt[maxn], to[maxn];

void AddEdge(int a, int b) {
    to[++m] = b; nxt[m] = head[a]; head[a] = m;
    return ;
}

int bel[maxn], siz[maxn], add[maxn], sum[maxn], indi[maxn];
void build(int u) {
    for(int e = head[u]; e; e = nxt[e]) if(to[e] != fa[u])
        bel[to[e]] = u, siz[u]++, build(to[e]);
    return ;
}

int main() {
    n = read(); q = read();
    rep(i, 2, n) fa[i] = read(), AddEdge(fa[i], i);

    bel[1] = 0; siz[0] = 1;
    build(1);
    int Ans = 0;
    rep(i, 1, q) {
        int x = read();
        sum[x] += siz[x];
        if(sum[x] >= MOD) sum[x] -= MOD;
        add[x]++;
        indi[x]++; sum[bel[x]]++; if(sum[bel[x]] >= MOD) sum[bel[x]] -= MOD;
        if(fa[x]) {
            indi[fa[x]]++; sum[bel[fa[x]]]++;
            if(sum[bel[fa[x]]] >= MOD) sum[bel[fa[x]]] -= MOD;
        }
        int nowans = (sum[x] + indi[x] + add[bel[x]]) % MOD;
        if(fa[x]) (nowans += indi[fa[x]] + add[bel[fa[x]]]) %= MOD;
        Ans += (LL)nowans * i % MOD;
        if(Ans >= MOD) Ans -= MOD;
    }

    printf("%d\n", Ans);

    return 0;
}
全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务