POJ 3275 Ranking the Cows floyd求传递闭包 || bfs

题目：

http://poj.org/problem?id=3275

题意：

有n个奶牛，有m个关系，形式如x y即x > y，现在给出的关系可能是不全的导致无法排出所有奶牛的名次，问最少添加再添加多个关系可以在任何情况都能排出所有奶牛的名次

思路

很明显的传递闭包，用邻接矩阵实现的floyd超时了，然后就写了一个bfs，300ms过了，然后看了一下大家的写法，基本都是用邻接表优化的floyd实现的，快的飞起，赶紧学了一发。之所以可以用邻接表优化，是因为枚举中间点k的时候，对于点i到点j，若是i到k或者k到j无边，是绝对无法更新的，所以枚举中间点k之后，只枚举和k有边相连的i和j，这题的边又比较少，复杂度就可以接受了
邻接表：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 1000 + 10;

bool mp[N][N];
int arr[N][N], brr[N][N];
int ka[N], kb[N];

int main()
{
    int n, m;
    while(~ scanf("%d%d", &n, &m))
    {
        memset(mp, 0, sizeof mp);
        memset(ka, 0, sizeof ka);
        memset(kb, 0, sizeof kb);
        int v, u;
        for(int i = 1; i <= m; i++)
        {
            scanf("%d%d", &v, &u);
            mp[v][u] = true;
            arr[v][++ka[v]] = u;
            brr[u][++kb[u]] = v;
        }
        for(int k = 1; k <= n; k++)
        {
            for(int i = 1; i <= kb[k]; i++)
            {
                int v = brr[k][i];
                for(int j = 1; j <= ka[k]; j++)
                {
                    int u = arr[k][j];
                    if(! mp[v][u])
                    {
                        mp[v][u] = true;
                        arr[v][++ka[v]] = u;
                        brr[u][++kb[u]] = v;
                    }
                }
            }
        }
        int ans = 0;
        for(int i = 1; i <= n; i++) ans += ka[i] + kb[i];
        ans = n * (n-1) / 2 - ans/2;//若是全序，那么应该有C(n,2)个关系，减去已有的，就是需要的
        printf("%d\n", ans);
    }
    return 0;
}

链式前向星实现的邻接表：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 1000 + 10;

struct edge
{
    int to, next;
}g[N*N*2];

int cnt, head[N], rhead[N];

bool mp[N][N];

void add_edge(int v, int u)
{
    g[cnt].to = u, g[cnt].next = head[v], head[v] = cnt++;
}
void radd_edge(int v, int u)
{
    g[cnt].to = u, g[cnt].next = rhead[v], rhead[v] = cnt++;
}
int main()
{
    int n, m;
    while(~ scanf("%d%d", &n, &m))
    {
        memset(mp, 0, sizeof mp);
        memset(head, -1, sizeof head);
        memset(rhead, -1, sizeof rhead);
        int v, u;
        for(int i = 1; i <= m; i++)
        {
            scanf("%d%d", &v, &u);
            mp[v][u] = true;
            add_edge(v, u);
            radd_edge(u, v);
        }
        for(int k = 1; k <= n; k++)
        {
            for(int i = rhead[k]; i != -1; i = g[i].next)
            {
                int v = g[i].to;
                for(int j = head[k]; j != -1; j = g[j].next)
                {
                    int u = g[j].to;
                    if(! mp[v][u])
                    {
                        mp[v][u] = true;
                        add_edge(v, u);
                        radd_edge(u, v);
                    }
                }
            }
        }
        int ans = 0;
        for(int i = 1; i <= n; i++)
            for(int j = i+1; j <= n; j++)
                if(!mp[i][j] && !mp[j][i]) ans++;//直接统计还需要的关系数量
        printf("%d\n", ans);
    }
    return 0;
}

bfs：
两个点之间没有任何联系，说明应该在两个点之间添加一条关系，可以用bfs统计和当前点没有任何联系的点的数量

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>

using namespace std;

const int N = 1000 + 10;

struct node
{
    int to, next;
}g[N*20];

int cnt, head[N];
int a[N*20], b[N*20], num[N];
bool vis[N];
void add_edge(int v, int u)
{
    g[cnt].to = u, g[cnt].next = head[v], head[v] = cnt++;
}
int bfs(int s)
{
    queue<int> que;
    memset(vis, 0, sizeof vis);
    que.push(s), vis[s] = true;
    int tot = 0;
    while(! que.empty())
    {
        int v = que.front(); que.pop();
        for(int i = head[v]; i != -1; i = g[i].next)
        {
            int u = g[i].to;
            if(! vis[u]) que.push(u), vis[u] = true, tot++;
        }
    }
    return tot;
}

int main()
{
    int n, m;
    while(~ scanf("%d%d", &n, &m))
    {
        cnt = 0;
        memset(head, -1, sizeof head);
        memset(num, 0, sizeof num);
        for(int i = 1; i <= m; i++)
        {
            scanf("%d%d", &a[i], &b[i]);
            add_edge(a[i], b[i]);
        }
        for(int i = 1; i <= n; i++) num[i] += bfs(i);//正向建图，对每个点bfs，求出从当前点能到达的点的数量
        cnt = 0;
        memset(head, -1, sizeof head);
        for(int i = 1; i <= m; i++) add_edge(b[i], a[i]);
        for(int i = 1; i <= n; i++) num[i] += bfs(i);//反向建图，对每个点bfs，求出从当前点能到达的点的数量
        int ans = 0;
        for(int i = 1; i <= n; i++) ans += n-1 - num[i];
        printf("%d\n", ans / 2);
    }
    return 0;
}