题目链接:
HDU 2647 Reward
题意:
有
n
个人和
数据范围:
n≤10000,m≤20000
分析;
把队列中的每个元素设有两个属性(
pair
):编号和奖金就好了。除了第一次0度的元素进队列,其余每次进队列奖金值+1就好了。
#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <algorithm>
#include <climits>
#include <cmath>
#include <ctime>
#include <cassert>
#include <vector>
#include <queue>
#define IOS ios_base::sync_with_stdio(0); cin.tie(0);
using namespace std;
typedef long long ll;
const int MAX_N = 10010;
vector<int> vec[MAX_N];
queue<pair<int, int > > que;
int degree[MAX_N];
void TopoSort(int n, int m)
{
while(!que.empty()) que.pop();
ll ans = 0;
for(int i = 1; i <= n; ++i) {
if(degree[i] == 0) {
que.push(make_pair(i, 888));
}
}
while(!que.empty()) {
pair<int, int> cur = que.front();
que.pop();
int u = cur.first, w = cur.second;
ans += w;
for(int i = 0; i < vec[u].size(); ++i) {
int to = vec[u][i];
degree[to]--;
if(degree[to] == 0) {
que.push(make_pair(to, w + 1));
}
}
}
for(int i = 1; i <= n; ++i) {
if(degree[i]) {
ans = -1;
break;
}
}
printf("%lld\n", ans);
}
int main()
{
int n, m;
while(~scanf("%d%d", &n, &m)) {
memset(degree, 0, sizeof(degree));
for(int i = 1; i <= n; ++i) vec[i].clear();
for(int i = 0; i < m; ++i) {
int u, v;
scanf("%d%d", &u, &v);
degree[u]++;
vec[v].push_back(u);
}
TopoSort(n, m);
}
return 0;
}