点权和

点权和

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;
}
全部评论

相关推荐

06-20 17:42
东华大学 Java
凉风落木楚山秋:要是在2015,你这简历还可以月入十万,可惜现在是2025,已经跟不上版本了
我的简历长这样
点赞 评论 收藏
分享
06-13 17:33
门头沟学院 Java
顺序不记了,大致顺序是这样的,有的相同知识点写分开了1.基本数据类型2.基本数据类型和包装类型的区别3.==和equals区别4.ArrayList与LinkedList区别5.hashmap底层原理,put操作时会发生什么6.说出几种树型数据结构7.B树和B+树区别8.jvm加载类机制9.线程池核心参数10.创建线程池的几种方式11.callable与runnable区别12.线程池怎么回收线程13.redis三剑客14.布隆过滤器原理,不要背八股,说说真正使用时遇到了问题没有(我说没有,不知道该怎么回答了)15.堆的内存结构16.自己在写项目时有没有遇见过oom,如何处理,不要背八股,根据真实经验,我说不会17.redis死锁怎么办,watchdog机制如何发现是否锁过期18.如何避免redis红锁19.一个表性别与年龄如何加索引20.自己的项目的QPS怎么测的,有没有真正遇到大数量表21.说一说泛型22.springboot自动装配原理23.springmvc与springboot区别24.aop使用过嘛?动态代理与静态代理区别25.spring循环依赖怎么解决26.你说用过es,es如何分片,怎么存的数据,1000万条数据怎么写入库中27.你说用limit,那么在数据量大之后,如何优化28.rabbitmq如何批次发送,批量读取,答了延迟队列和线程池,都不对29.计网知不知道smtp协议,不知道写了对不对,完全听懵了30.springcloud知道嘛?只是了解反问1.做什么的?短信服务,信息量能到千万级2.对我的建议,基础不错,但是不要只背八股,多去实际开发中理解。面试官人不错,虽然没露脸,但是中间会引导我回答问题,不会的也只是说对我要求没那么高。面完问我在济宁生活有没有困难,最快什么时候到,让人事给我聊薪资了。下午人事打电话,问我27届的会不会跑路,还在想办法如何使我不跑路,不想扣我薪资等。之后我再联系吧,还挺想去的😭,我真不跑路哥😢附一张河科大幽默大专图,科大就是大专罢了
查看30道真题和解析
点赞 评论 收藏
分享
不愿透露姓名的神秘牛友
07-24 13:36
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务