HDU - 4034 Graph （floyd变形）

题目大意：给出每个点之间的最短距离，问需要几条边才能形成这张图

解题思路：100个点，果断floyd，接着判断
如果dp[i][j] > dp[i][k] + dp[k][j]表示这张图构建不了，因为最短路冲突了
如果dp[i][j] == dp[i][k] + dp[k][j]，就表示ij这条路可以被ik和kj取代，那么这条路就可去掉了
去掉的路记得标记。。。WA了好多次

#include <cstdio>
#include <algorithm>
using namespace std;
#define N 110
int dis[N][N], cas = 1, n;
bool vis[N][N];

void init() {
    scanf("%d", &n);
    for (int i = 0; i < n; i++)
        for (int j = 0; j < n; j++) {
            scanf("%d", &dis[i][j]);
            vis[i][j] = 0;
        }
}

int floyd() {
    int ans = n * (n -1);
    for (int k = 0; k < n; k++)
        for (int i = 0; i < n; i++)
            for (int j = 0; j < n; j++) {
                if (dis[i][j] > dis[i][k] + dis[k][j]){
                    return -1;
                }
                if (i != k && k != j && i != j && !vis[i][j] && dis[i][j] == dis[i][k] + dis[k][j]) {
                    vis[i][j] = 1;
                    ans--;
                }
            }
    return ans;
}

void solve() {
    int ans = floyd();
    printf("Case %d: ", cas++);
    if (ans == -1)
        printf("impossible\n");
    else
        printf("%d\n", ans);
}

int main() {
    int test;
    scanf("%d", &test);
    while (test--) {
        init();
        solve();
    }
    return 0;
}