洛谷 P2341 [HAOI2006]受欢迎的牛
Description
每头奶牛都梦想成为牛棚里的明星。被所有奶牛喜欢的奶牛就是一头明星奶牛。所有奶
牛都是自恋狂,每头奶牛总是喜欢自己的。奶牛之间的“喜欢”是可以传递的——如果A喜
欢B,B喜欢C,那么A也喜欢C。牛栏里共有N 头奶牛,给定一些奶牛之间的爱慕关系,请你
算出有多少头奶牛可以当明星。
Input
第一行:两个用空格分开的整数:N和M
第二行到第M + 1行:每行两个用空格分开的整数:A和B,表示A喜欢B
Output
- 第一行:单独一个整数,表示明星奶牛的数量
Sample Input
3 3
1 2
2 1
2 3
Sample Output
1
Data Size
10%的数据N<=20, M<=50
30%的数据N<=1000,M<=20000
70%的数据N<=5000,M<=50000
100%的数据N<=10000,M<=50000
题解:
- n个点,n条边。明显有环。可以考虑tarjan缩点,然后跑一遍DAG。dp的思路就是设一个f[i]表示有多少人喜欢i,然后通过缩点转换成一个DAG就可以转移了。转移方程如下:
for(int i = 1; i <= sdn; i++) { int x = seq[i]; for(int i = h[x]; i != 0; i = e[i].next) f[x] += f[e[i].to]; f[x] += w[x]; if(f[x] == n) //TODO } //sdn为缩点之后点的个数 //seq为缩点之后的拓扑序 //w为这个缩完的点包含原来的点的个数
但,这样是有问题的。因为一个点可能会被重复计数。具体证明过程:这里
- 所以,我们要用一个非常巧妙的结论。在DAG里,如果有且只有一个点的出度为0的话,那么每个点都可以遍历到这个点。
画个图,很容易理解。如果有2个这样的点x、y就不行,证明很容易:x的爱慕无法传达给y。
#include <iostream>
#include <cstdio>
#include <stack>
#include <queue>
#define maxn 50005
using namespace std;
struct E {int next, to;} e[maxn];
int n, m, num;
int h[maxn], out[maxn];
int dex, tot;
int dfn[maxn], low[maxn], o[maxn], belong[maxn];
bool vis[maxn];
stack<int> stk;
int read()
{
int x = 0; char c = getchar();
while(c < '0' || c > '9') c = getchar();
while(c >= '0' && c <= '9') {x = x * 10 + c - '0'; c = getchar();}
return x;
}
void add(int u, int v)
{
e[++num].next = h[u];
e[num].to = v;
h[u] = num;
}
void tarjan(int x)
{
low[x] = dfn[x] = ++dex;
stk.push(x), vis[x] = 1;
for(int i = h[x]; i != 0; i = e[i].next)
{
int now = e[i].to;
if(!dfn[now])
tarjan(now),
low[x] = min(low[x], low[now]);
else if(vis[now])
low[x] = min(low[x], dfn[now]);
}
if(low[x] == dfn[x])
{
tot++;
while(1)
{
int now = stk.top();
stk.pop(); vis[now] = 0;
o[tot]++, belong[now] = tot;
if(now == x) break;
}
}
}
int main()
{
cin >> n >> m;
for(int i = 1; i <= m; i++)
{
int u = read(), v = read();
add(u, v);
}
for(int i = 1; i <= n; i++)
if(!dfn[i]) tarjan(i);
for(int i = 1; i <= n; i++)
for(int j = h[i]; j != 0; j = e[j].next)
{
int x = i, y = e[j].to;
if(belong[x] != belong[y])
out[belong[x]]++;
}
int sum = 0, pos;
for(int i = 1; i <= tot; i++)
if(!out[i]) sum++, pos = i;
if(sum != 1) cout << 0;
else cout << o[pos];
return 0;
}