1. 程式人生 > >P2597 [ZJOI2012]災難 拓撲排序

P2597 [ZJOI2012]災難 拓撲排序

out string 生產 意思 個數字 push 現在 我沒 node

這個題有點意思,正常寫法肯定會T,然後需要優化。先用拓撲排序重構一遍樹,然後進行一個非常神奇的操作:把每個點放在他的食物的lca上,然後計算的時候直接dfs全加上就行了。為什麽呢,因為假如你的食物的lca死了,你就很自然的死了。這個題還要加一個超級源點,一開始我沒加只拿了20分。但是不知道為什麽,後來想到有可能一個兄弟能吃兩個生產者,在查詢lca的時候會跪。。。剩下就沒啥了,順便練一下倍增求lca。

題幹:

題目描述

阿米巴是小強的好朋友。

阿米巴和小強在草原上捉螞蚱。小強突然想,如果螞蚱被他們捉滅絕了,那麽吃螞蚱的小鳥就會餓死,而捕食小鳥的猛禽也會跟著滅絕,從而引發一系列的生態災難。

學過生物的阿米巴告訴小強,草原是一個極其穩定的生態系統。如果螞蚱滅絕了,小鳥照樣可以吃別的蟲子,所以一個物種的滅絕並不一定會引發重大的災難。

我們現在從專業一點的角度來看這個問題。我們用一種叫做食物網的有向圖來描述生物之間的關系:

一個食物網有N個點,代表N種生物,如果生物x可以吃生物y,那麽從y向x連一個有向邊。

這個圖沒有環。

圖中有一些點沒有連出邊,這些點代表的生物都是生產者,可以通過光合作用來生存; 而有連出邊的點代表的都是消費者,它們必須通過吃其他生物來生存。

如果某個消費者的所有食物都滅絕了,它會跟著滅絕。

我們定義一個生物在食物網中的“災難值”為,如果它突然滅絕,那麽會跟著一起滅絕的生物的種數。

舉個例子:在一個草場上,生物之間的關系是:

如

如果小強和阿米巴把草原上所有的羊都給嚇死了,那麽狼會因為沒有食物而滅絕,而小強和阿米巴可以通過吃牛、牛可以通過吃草來生存下去。所以,羊的災難值是1。但是,如果草突然滅絕,那麽整個草原上的5種生物都無法幸免,所以,草的災難值是4。

給定一個食物網,你要求出每個生物的災難值。
輸入輸出格式
輸入格式:

輸入文件 catas.
in 的第一行是一個正整數 N,表示生物的種數。生物從 1 標 號到 N。 接下來 N 行,每行描述了一個生物可以吃的其他生物的列表,格式為用空 格隔開的若幹個數字,每個數字表示一種生物的標號,最後一個數字是 0 表示列 表的結束。 輸出格式: 輸出文件catas.out包含N行,每行一個整數,表示每個生物的災難值。

代碼:

#include<iostream>
#include<cstdio>
#include<cmath>
#include<ctime>
#include<queue>
#include
<algorithm> #include<cstring> using namespace std; #define duke(i,a,n) for(register int i = a;i <= n;++i) #define lv(i,a,n) for(register int i = a;i >= n;--i) #define clean(a) memset(a,0,sizeof(a)) const int INF = 1 << 30; typedef long long ll; typedef double db; template
<class T> void read(T &x) { char c; bool op = 0; while(c = getchar(), c < 0 || c > 9) if(c == -) op = 1; x = c - 0; while(c = getchar(), c >= 0 && c <= 9) x = x * 10 + c - 0; if(op) x = -x; } template <class T> void write(T x) { if(x < 0) putchar(-), x = -x; if(x >= 10) write(x / 10); putchar(0 + x % 10); } const int N = 1e5 + 5; struct node { int l,r,nxt; }a[N],b[N],c[N]; int len = 0,len1 = 0,len2 = 0,n; int in[N],rk[N],num = 0; int lst[N],lst1[N],lst2[N]; int f[N][35],ans[N],h[N]; void add(int x,int y) { a[++len].l = x; a[len].r = y; a[len].nxt = lst[x]; lst[x] = len; } void add1(int x,int y) { b[++len1].l = x; b[len1].r = y; b[len1].nxt = lst1[x]; lst1[x] = len1; } void add2(int x,int y) { c[++len2].l = x; c[len2].r = y; c[len2].nxt = lst2[x]; lst2[x] = len2; } void topsort() { queue <int> q; duke(i,1,n) { if(!in[i]) q.push(i); } int num = 0; while(!q.empty()) { int now = q.front(); q.pop(); rk[++num] = now; for(int k = lst[now];k;k = a[k].nxt) { int y = a[k].r; in[y]--; if(!in[y]) q.push(y); } } } int lca(int x,int y) { if(h[x] < h[y]) { swap(x,y); } int k = h[x] - h[y]; for(int i = 0;i <= 30;i++) { if((k >> i) & 1) x = f[x][i]; } if(x == y) return x; for(int i = 30;i >= 0;i--) { if(f[x][i] != f[y][i]) { x = f[x][i]; y = f[y][i]; } } return f[x][0]; } void dfs(int now) { for(int k = lst2[now];k;k = c[k].nxt) { int y = c[k].r; dfs(y); ans[now] += ans[y]; } ans[now]++; } int main() { read(n); duke(i,1,n) { int x; read(x); while(x) { in[i]++; add1(i,x); add(x,i); read(x); } } topsort(); duke(i,1,n) { int x = b[lst1[rk[i]]].r; for(int k = lst1[rk[i]];k;k = b[k].nxt) { int y = b[k].r; x = lca(x,y); } add2(x,rk[i]); h[rk[i]] = h[x] + 1; f[rk[i]][0] = x; for(int j = 1;j <= 30;j++) f[rk[i]][j] = f[f[rk[i]][j - 1]][j - 1]; } dfs(0); duke(i,1,n) { printf("%d\n",ans[i] - 1); } return 0; }

P2597 [ZJOI2012]災難 拓撲排序