链接:第十六届上海大学程序设计联赛春季赛暨上海高校金马五校赛:B-合约数
题意:给出一个树,树的节点编号1-N,每个节点有个val,定义F(i) = 节点i的子树的节点中的val是val[i]的合约数的个数(包括节点i)。合约数:若x是y的约数,且x是合数,则称x是y的合约数。求 对1e9+7取模后的结果。
题解:DFS。直接遍历树,遍历时标记非合数节点的val的个数,然后逆求该节点对其祖先节点的贡献。注意在节点出递归栈消除标记。
#include <bits/stdc++.h>
using namespace std;
const double EPS = 1e-8;
const int mod = 1e9 + 7;
const int INF = 0x3f3f3f3f;
const int maxn = 20000 + 10;
const int maxv = 10000 + 10;
int n, p;
int val[maxn];
bool isprime[maxn], vis[maxn];
vector<int> g[maxn];
long long fa[maxn], ans = 0;
void sieve()
{
memset(isprime, false, sizeof(isprime));
isprime[0] = isprime[1] = false;
for(int i = 2; i < maxv; i++){
if(!isprime[i]){
for(int j = i << 1; j < maxv; j += i) isprime[j] = true;
}
}
}
void DFS(int v)
{
vis[v] = true;
if(isprime[val[v]]){
fa[val[v]] = (fa[val[v]] + v) % mod;
for(int i = 1; i * val[v] < maxv; i++) ans = (ans + fa[val[v] * i]) % mod;
}
for(int i = 0, n = g[v].size(); i < n; i++) if(!vis[g[v][i]]) DFS(g[v][i]);
if(isprime[val[v]]){
fa[val[v]] = (fa[val[v]] - v) % mod;
if(fa[val[v]] < 0) fa[val[v]] = (fa[val[v]] + mod) % mod;
}
}
int main()
{
sieve();
int T;
scanf("%d", &T);
while(T--){
scanf("%d%d", &n, &p);
for(int i = 1; i <= n; i++) g[i].clear();
memset(fa, 0, sizeof(fa));
memset(vis, false, sizeof(vis));
for(int i = 1, a, b; i < n; i++){
scanf("%d%d", &a, &b);
g[a].push_back(b);
g[b].push_back(a);
}
for(int i = 1; i <= n; i++) scanf("%d", &val[i]);
ans = 0;
DFS(p);
printf("%d\n", ans);
}
return 0;
}