HDOJ-----5695Gym Class拓扑（vector）

Gym Class

Time Limit: 6000/1000 MS (Java/Others)    Memory Limit: 65536/65536 K (Java/Others)
Total Submission(s): 1149    Accepted Submission(s): 474

Problem Description

众所周知，度度熊喜欢各类体育活动。

今天，它终于当上了梦寐以求的体育课老师。第一次课上，它发现一个有趣的事情。在上课之前，所有同学要排成一列， 假设最开始每个人有一个唯一的ID，从1到N，在排好队之后，每个同学会找出包括自己在内的前方所有同学的最小ID，作为自己评价这堂课的分数。麻烦的是，有一些同学不希望某个（些）同学排在他（她）前面，在满足这个前提的情况下，新晋体育课老师——度度熊，希望最后的排队结果可以使得所有同学的评价分数和最大。

 

Input

第一行一个整数T，表示T(1≤T≤30) 组数据。

对于每组数据，第一行输入两个整数N和M(1≤N≤100000,0≤M≤100000)，分别表示总人数和某些同学的偏好。

接下来M行，每行两个整数A 和B(1≤A,B≤N)，表示ID为A的同学不希望ID为B的同学排在他（她）之前。你可以认为题目保证至少有一种排列方法是符合所有要求的。

 

Output

对于每组数据，输出最大分数 。

 

Sample Input

3
1 0
2 1
1 2
3 1
3 1

 

Sample Output

1
2
6

拓扑排序，加上一个排序的优先队列就行了，开始以为要反向建边，后来发现需要从小到大严格拓扑的才要反向建边

vector版本

#include<cstdio>
#include<cstring>
#include<vector>
#include<queue>
#include<functional>
#include<algorithm>
#define maxn 100010
#define ll long long
#define inf 0x3f3f3f3f
#define mes(a, b) memset(a, b, sizeof(a))
using namespace std;
vector<int > edge[maxn];
int in[maxn];
int cnt, n, m;
ll ans;
void topo(){
    priority_queue<int > Q;
    for(int i = 1; i <= n; i++){
        if(!in[i]){
            Q.push(i);
        }
    }
    int c = inf;
    while(!Q.empty()){
        int u = Q.top();
        Q.pop();
        in[u] = -1;
        if(c > u){
            c = u;
        }
        ans += c;
        for(int i = 0; i < edge[u].size(); i++){
            int v = edge[u][i];
            in[v]--;
            if(!in[v]){
                Q.push(v);
            }
        }
    }
}
int main(){
    int t, a, b;
    scanf("%d", &t);
    while(t--){
        scanf("%d%d", &n, &m);
        mes(in, 0);
        for(int i = 1; i <= n; i++){
            edge[i].clear();
        }
        for(int i = 1; i <= m; i++){
            scanf("%d%d", &a, &b);
            edge[a].push_back(b);
            in[b]++;
        }
        ans = 0;
        topo();
        printf("%lld\n", ans);
    }
    return 0;
}

邻接表版本（比vector略快，但有时候vector方便，具体问题具体分析）

#include<cstdio>
#include<cstring>
#include<vector>
#include<queue>
#include<functional>
#include<algorithm>
#define maxn 100010
#define ll long long
#define inf 0x3f3f3f3f
#define mes(a, b) memset(a, b, sizeof(a))
using namespace std;
int in[maxn], head[maxn];
int n, m, num;
ll ans;
struct node{
    int to, next;
}edge[maxn*2];
void add(int u, int v){
    edge[num].to = v;
    edge[num].next = head[u];
    head[u] = num++;
    in[v]++;
}
void topo(){
    priority_queue<int > Q;
    for(int i = 1; i <= n; i++){
        if(!in[i]){
            Q.push(i);
        }
    }
    int c = inf;
    while(!Q.empty()){
        int u = Q.top();
        Q.pop();
        in[u] = -1;
        if(c > u){
            c = u;
        }
        ans += c;
        for(int i = head[u]; i != -1; i = edge[i].next){
            int v = edge[i].to;
            in[v]--;
            if(!in[v]){
                Q.push(v);
            }
        }
    }
}
int main(){
    int t, a, b;
    scanf("%d", &t);
    while(t--){
        scanf("%d%d", &n, &m);
        mes(in, 0); mes(head, -1);
        num = 0;
        for(int i = 1; i <= m; i++){
            scanf("%d%d", &a, &b);
            add(a, b);
        }
        ans = 0;
        topo();
        printf("%lld\n", ans);
    }
    return 0;
}