标签：灾难 int ZJOI2012 MAXN lca isdigit include getchar

Description

[ZJOI2012]灾难

Solution

我们可
反向建图

Code

#include<iostream>
#include<cstdio>
#include<queue>
using namespace std;
#define MAXN 1000010
struct rec {
    int nxt, ver;
} t[MAXN];
int n, cnt, s;
int head[MAXN], in[MAXN], size[MAXN], d[MAXN], lca[MAXN][21], h[MAXN];
queue <int> q;
inline int read() {
    int s = 0, w = 1;
    char c = getchar();
    for (; !isdigit(c); c = getchar()) if (c == '-') w = -1;
    for (; isdigit(c); c = getchar()) s = (s << 1) + (s << 3) + (c ^ 48);
    return s * w;
}
inline void add(int u, int v) {
    t[++cnt].nxt = head[u], t[cnt].ver = v, head[u] = cnt;
}
inline void TopSort() {
    for (register int i = 1; i <= n; i++)
        if (!in[i]) q.push(i);
    while (!q.empty()) {
        int u = q.front();
        q.pop();
        h[++s] = u;
        for (register int i = head[u]; i; i = t[i].nxt) {
            int v = t[i].ver;
            if (!(--in[v])) q.push(v);
        }
    }
}
inline int LCA(int u, int v) {
    if (d[u] < d[v]) swap(u, v);
    for (register int i = 20; i >= 0; i--)
        if (d[lca[u][i]] >= d[v]) u = lca[u][i];
    if (u == v) return u;
    for (register int i = 20; i >= 0; i--)
        if (lca[u][i] != lca[v][i]) u = lca[u][i], v = lca[v][i];
    return lca[u][0];
}
int main() {
    n = read();
    for (register int i = 1; i <= n; i++) {
        int x = read();
        while (x) ++in[x], add(i, x), x = read();
    }
    TopSort();
    for (register int i = n; i; i--) {
        int u = h[i], x = t[head[u]].ver;
        for (register int j = t[head[u]].nxt; j; j = t[j].nxt) {
            int v = t[j].ver;
            x = LCA(x, v);
        }
        lca[u][0] = x, d[u] = d[x] + 1;
        for (register int j = 1; j <= 20; j++) lca[u][j] = lca[lca[u][j - 1]][j - 1];
    }
    for (register int i = 1; i <= n; i++) ++size[h[i]], size[lca[h[i]][0]] += size[h[i]];
    for (register int i = 1; i <= n; i++) printf("%d\n", size[i] - 1);
    return 0;
}

标签：灾难,int,ZJOI2012,MAXN,lca,isdigit,include,getchar
来源： https://www.cnblogs.com/Agakiss/p/11621043.html

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