题意:
有一个比赛,一共 N 个选手,已知 M 种胜负关系,且胜负关系可以传递(若 A 胜 B,B 胜 C,则 A 胜 B)。求:有几对选手的胜负关系无法预测。
输入要求:
第 1 行:输入数据组数。
每组数据第 1 行:输入两个数 N 和 M,N 为选手个数,M 为已知胜负关系个数;
每组数据第 2 行至第 M + 1 行:输入两个数 A,B,表示 A 胜 B。
输出要求:
每组数据输出一行一个数 ans,表示无法预测胜负的选手对数。
Sample Input:
3
3 3
1 2
1 3
2 3
3 2
1 2
2 3
4 2
1 2
3 4
Sample Output:
0
0
4
思路:
- 构造一个二维数组 dis[N][N],dis[i][j] = 1表示选手 i 胜选手 j,dis[i][j] = 0 表示选手 i 和 j 的胜负关系未知。初始化对已知胜负选手的 dis 赋值。
- 初始化之后,使用 Floyd 算法,若 dis[k][j] = 1 且 dis[j][l] = 1,则将 dis[k][l] 也赋值为1。
- 再对 Floyd 算法进行剪枝:在第二重循环中,若 dis[k][j] != 1 且 dis[j][k] != 1,则不需要进入第三重循环。
- Floyd 结束之后,使用二重循环,枚举判断每对选手胜负情况是否已知,得到答案。
代码:
#include <iostream>
using namespace std;
int a[505][505];
int dis[505][505];
int main()
{
int n;//组数
cin >> n;
for (int i = 0; i < n; i++) {
int N, M;
cin >> N >> M;
for (int j = 1; j <= N; j++) {
for (int k = 1; k <= N; k++) {
dis[j][k] = 0;
}
}
int A, B;
for (int j = 0; j < M; j++) {
cin >> A >> B;
dis[A][B] = 1;
}
for (int j = 1; j <= N; j++) {
for (int k = 1; k <= N; k++) {
if (dis[j][k] == 1 || dis[k][j] == 1) {
for (int l = k + 1; l <= N; l++) {
if (dis[k][j] == 1 && dis[j][l] == 1)
dis[k][l] = 1;
if (dis[j][k] == 1 && dis[l][j] == 1)
dis[l][k] = 1;
}
}
}
}
int ans = 0;
for (int j = 1; j <= N - 1; j++) {
for (int k = j + 1; k <= N; k++) {
if (dis[j][k] == 0 && dis[k][j] == 0)
ans++;
}
}
cout << ans << endl;
}
}