第十六届上海大学程序设计联赛春季赛暨上海高校金马五校赛：B-合约数（DFS）

链接：第十六届上海大学程序设计联赛春季赛暨上海高校金马五校赛：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;
}