第十六届金马五校 B:

14人阅读 评论(0) 收藏 举报
分类:
链接:https://www.nowcoder.com/acm/contest/91/B
来源:牛客网

题目描述

在埃森哲,员工培训是最看重的内容,最近一年,我们投入了 9.41 亿美元用于员工培训和职业发展。截至 2018 财年末,我们会在全球范围内设立 100 所互联课堂,将互动科技与创新内容有机结合起来。按岗培训,按需定制,随时随地,本土化,区域化,虚拟化的培训会让你快速取得成长。小埃希望能通过培训学习更多ACM 相关的知识,他在培训中碰到了这样一个问题,

给定一棵n个节点的树,并且根节点的编号为p,第i个节点有属性值vali, 定义F(i): 在以i为根的子树中,属性值是vali的合约数的节点个数。y 是 x 的合约数是指 y 是合数且 y 是 x 的约数。小埃想知道1000000007取模后的结果.

输入描述:

输入测试组数T,每组数据,输入n+1行整数,第一行为n和p,1<=n<=20000, 1<=p<=n, 接下来n-1行,每行两个整数u和v,表示u和v之间有一条边。第n+1行输入n个整数val1, val2,…, valn,其中1<=vali<=10000,1<=i<=n.

输出描述:

对于每组数据,输出一行,包含1个整数, 表示对1000000007取模后的结果

示例1

输入

2
5 4
5 3
2 5
4 2
1 3
10 4 3 10 5
3 3
1 3
2 1
1 10 1

输出

11
2

备注:

n>=10000的有20组测试数据

题解:

首先预处理1-N里的每个数的合约数,复杂度O(N*sqrt(N)), 然后根据DFS序记录目前为止每个数出现的次数,第一次到这个节点i时候先F[i]先减去他之前合约数出现的总次数,再次回溯的时候再次加上他合数出现的次数,两次得到的即为F[i].

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
const int N = 20100;

vector<int>g[N];
int val[N];
bool isprime[N];
vector<short>f[N];
int cnt[N], F[N];
ll res;
const ll MOD = 1E9 + 7;
void init()
{
    for(int i = 2;i < N;i ++) isprime[i] = 1;
    for(int i = 2;i < N;i ++) {
        if(isprime[i]) {
            for(int j = i + i;j < N;j += i) {
                isprime[j] = 0;
            }
        }
    }
    for(int i = 4;i <= 10000;i ++) {
        if(!isprime[i]) {
            for(int j = i;j <= 10000;j += i) {
                f[j].push_back(i);
            }
        }
    }
}
void dfs(int u, int par)
{
    for(int i = 0;i < f[val[u]].size();i ++) {
        int num = f[val[u]][i];
        F[u] -= cnt[num]; //根据dfs序,先减去他之前出现的
    }
    cnt[val[u]] ++;
    for(int i = 0;i < g[u].size();i ++) {
        int v = g[u][i];
        if(v == par) continue;
        dfs(v, u);
    }
    for(int i = 0;i < f[val[u]].size();i ++) {
        int num = f[val[u]][i];
        F[u] += cnt[num]; //回溯到他时,+现在出现过的总数 = 遍历完其子树后的总合约数个数-之前出现的合约数个数.
    }
    res += F[u] * 1LL * u;
    res %= MOD;
}
int main()
{
    int T;
    init();
    scanf("%d", &T);
    while(T --) {
        memset(cnt, 0, sizeof(cnt));
        res = 0;
        int n, p;
        scanf("%d %d", &n, &p);
        for(int i = 1;i <= n;i ++) F[i] = 0,g[i].clear();
        for(int i = 1;i < n;i ++) {
            int u, v;
            scanf("%d %d", &u, &v);
            g[u].push_back(v);
            g[v].push_back(u);
        }
        for(int i = 1;i <= n;i ++) {
            scanf("%d", &val[i]);
        }
        dfs(p,-1);
        printf("%lld\n", res);
    }
    return 0;
}


查看评论

编程方向

 一、前言  很多人认为,数据库编程是一个颇具诱惑力的工作,这诱惑力来自于数据库应用的巨大市场需求。计算机应用有科学计算、数据处理与过程控制三大主要领域,而数据处理是其中所占比重最大的一个领域,包括现...
  • xingjiaren
  • xingjiaren
  • 2001-04-10 11:37:00
  • 924

埃森哲杯第十六届上海大学程序设计联赛春季赛暨上海高校金马五校赛 J-小Y写文章(二分+二分图匹配)

链接:https://www.nowcoder.com/acm/contest/91/J来源:牛客网题目描述 小Y写了一篇文章,他对自己的文笔很有自信,尤其是自己总结出了一套计算文章通顺性的公式。 ...
  • albertluf
  • albertluf
  • 2018-04-18 10:27:07
  • 83

埃森哲杯第十六届上海大学程序设计联赛春季赛暨上海高校金马五校赛 F-1+2=3?

链接:https://www.nowcoder.com/acm/contest/91/F来源:牛客网题目描述 小Y在研究数字的时候,发现了一个神奇的等式方程,他屈指算了一下有很多正整数x满足这个等式...
  • albertluf
  • albertluf
  • 2018-04-15 16:46:44
  • 135

第15届上海大学程序设计联赛夏季赛暨上海高校金马五校赛(部分题解)

唉,好久没更新博客了,说明我这段时间在学习上都懈怠了Σ( ° △ °|||)︴昨天打了场金马赛,这网络…..我真的是无力吐槽…..前三题都是大水题,就不挂代码了;D-快速幂取模裸题#include u...
  • Murphyc
  • Murphyc
  • 2017-07-10 13:54:47
  • 484

埃森哲杯第十六届上海大学程序设计联赛春季赛暨上海高校金马五校赛F题 1+2=3?

题目链接:https://www.nowcoder.com/acm/contest/91/F题意:求第N个符合条件X^2X=3X的X。N&amp;lt;=10的12次方。解题思路:我刚开始想到最粗暴的...
  • weixin_36416680
  • weixin_36416680
  • 2018-04-16 11:01:00
  • 61

金马五校赛 I : 二数

题目链接:点击打开链接大意:给定一个十进制下最多105位的数字,请你求出和这个数字的差的绝对值最小的二数,若答案不唯一,输出最小的那个。 也就是说,给定数字n,求出m,使得abs(n-m)最小且...
  • swunHJ
  • swunHJ
  • 2018-04-15 20:19:14
  • 18

【牛客网】【埃森哲杯第十六届上海大学程序设计联赛春季赛暨上海高校金马五校赛】B—合约数【莫队做法】

题目链接:B—合约数题意:一棵树,有n个节点,从1编号到n。根节点的编号为p。给出每个节点的val[i]值,定义f(i)为以编号i为根节点的子树中(包括根节点),所有val[j]是合数并且是val[i...
  • gymgym1212
  • gymgym1212
  • 2018-04-15 20:33:28
  • 137

【牛客网】【埃森哲杯第十六届上海大学程序设计联赛春季赛暨上海高校金马五校赛】L—K序列【动态规划】

题目链接:L—K序列题解:for一遍数组,temp[i]表示:不包括当前for到的点,序列和为i的最长长度。然后for一遍0-k,对dp[ (j+num[i])%k ]进行更新,dp[j]是包括当前点...
  • gymgym1212
  • gymgym1212
  • 2018-04-15 18:46:39
  • 174

埃森哲杯第十六届上海大学程序设计联赛春季赛暨上海高校金马五校赛H题小Y与多米诺骨牌(线段树优化dp)

题意 题目链接:https://www.nowcoder.com/acm/contest/91/H 来源:牛客网 题解 设l[i]l[i]l[i]为向左推第iii个骨牌最远能影响到的骨牌的...
  • ME495
  • ME495
  • 2018-04-17 11:20:36
  • 89

第十六届上海大学程序设计联赛春季赛暨上海高校金马五校赛 B 合约数(启发式合并)

链接:https://www.nowcoder.com/acm/contest/91/B来源:牛客网题目描述 在埃森哲,员工培训是最看重的内容,最近一年,我们投入了 9.41 亿美元用于员工培训和职...
  • lzc504603913
  • lzc504603913
  • 2018-04-15 19:17:02
  • 54
    个人资料
    持之以恒
    等级:
    访问量: 5642
    积分: 1422
    排名: 3万+