标签：长链 Invitational 剖分 int len son dfs now id

 题库链接

考虑莫比乌斯， 套上去之后就是变成了统计长度为d的一共有多少路径， 直接长链剖分， 

在计蒜客上极度卡常， 卡了一万年才卡过去， 现场好像还有用点分治过去的， 这都能过？？

#include<bits/stdc++.h>
#define LL long long
using namespace std;

const int N = (int)5e5 + 7;
const int M = 30000;

int n, d, a[N], vis[N], miu[M + 1];
int now_val, now_op, now_cnt, now_col;
int len[N], son[N], dp[N], *id, *f[N];
bool ok[N];
LL ans;

vector<int> V[M + 1];
vector<int> P[M + 1];

int edge_tot, head[N];
struct Edge {
    int to, nex;
} e[N << 1];

inline void addEdge(int u, int v) {
    e[edge_tot].to = v;
    e[edge_tot].nex = head[u];
    head[u] = edge_tot++;
}

bool checkNum(int x) {
    for(int i = 2; i * i <= x; i++) {
        if(x % (i * i) == 0) return false;
    }
    return true;
}

void prepare() {
    ok[1] = true;
    for(int i = 2; i <= M; i++) {
        if(checkNum(i)) ok[i] = true;
    }
    for(int i = 1; i <= M; i++) {
        for(int j = 1; j * j <= i; j++) {
            if(i % j) continue;
            if(ok[j]) V[i].push_back(j);
            if(j * j != i && ok[i / j]) V[i].push_back(i / j);
        }
    }
    miu[1] = 1;
    for(int i = 1; i <= M; i++) {
        for(int j = i + i; j <= M; j += i) {
            miu[j] -= miu[i];
        }
    }
}

void gao(int u, int fa) {
    now_cnt++;
    vis[u] = now_col;
    son[u] = len[u] = 0;
    for(int j = head[u], v; ~j; j = e[j].nex) {
        v = e[j].to;
        if(v == fa || a[v] % now_val) continue;
        gao(v, u);
        if(len[v] > len[u]) {
            len[u] = len[v];
            son[u] = v;
        }
    }
    len[u]++;
}

void dfs(int u, int fa) {
    f[u][0] = 1;
    if(son[u]) {
        f[son[u]] = f[u] + 1;
        dfs(son[u], u);
    }
    if(d < len[u]) {
        ans += now_op * f[u][d];
    }
    for(int j = head[u], v; ~j; j = e[j].nex) {
        v = e[j].to;
        if(v == fa || v == son[u] || a[v] % now_val) continue;
        f[v] = id; id += len[v];
        dfs(v, u);
        for(int i = 1; i <= len[v] && i <= d; i++) {
            if(d - i < len[u]) ans += 1LL * now_op * f[v][i - 1] * f[u][d - i];
        }
        for(int i = 1; i <= len[v] && i <= d; i++) {
            f[u][i] += f[v][i - 1];
        }
    }
}

void init() {
    ans = edge_tot = 0;
    for(int i = 1; i <= M; i++) {
        P[i].clear();
    }
    for(int i = 1; i <= n; i++) {
        head[i] = -1;
        vis[i] = 0;
    }
}

int main() {
    prepare();
    int cas = 0;
    int T;
    scanf("%d", &T);
    while(T--) {
        scanf("%d%d", &n, &d);
        init();
        for(int i = 1; i <= n; i++) {
            scanf("%d", &a[i]);
            for(auto &t : V[a[i]]) {
                P[t].push_back(i);
            }
        }
        for(int i = 1; i < n; i++) {
            int u, v;
            scanf("%d%d", &u, &v);
            addEdge(u, v);
            addEdge(v, u);
        }
        for(int i = 2; i <= M; i++) {
            if(!P[i].size()) continue;
            now_val = i; now_op = -miu[i]; now_col = i;
            for(auto &Rt : P[i]) {
                if(vis[Rt] == i) continue;
                now_cnt = 0;
                gao(Rt, 0);
                for(int i = 0; i <= now_cnt; i++) dp[i] = 0;
                id = dp;
                f[Rt] = id; id += len[Rt];
                dfs(Rt, 0);
            }
        }
        printf("Case #%d: %lld\n", ++cas, 2 * ans);
    }
    return 0;
}

/*
*/

标签：长链,Invitational,剖分,int,len,son,dfs,now,id
来源： https://www.cnblogs.com/CJLHY/p/11635002.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。