[最小生成树+flyod] 牛客第六场 I

题目

题目链接：https://ac.nowcoder.com/acm/contest/3007/I

思路

不知道说啥思路
这道题就是先跑一次线段树看看建出来的图是不是和输入一样

代码

#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<cctype>
#include<ctime>
#include<iostream>
#include<string>
#include<map>
#include<queue>
#include<stack>
#include<set>
#include<vector>
#include<iomanip>
#include<list>
#include<bitset>
#include<sstream>
#include<fstream>
#include<complex>
#include<algorithm>
#if __cplusplus >= 201103L
#include <unordered_map>
#include <unordered_set>
#endif
#define ll long long
using namespace std;
const int INF = 0x3f3f3f3f;
int b[600][600],c[600][600];
int f[600];
struct sut{
	int x,y,m;
}a[1000010];
int cmp(sut a,sut b){
	return a.m<b.m;
}
int find(int x){
	if(f[x]==x) return x;
	else return f[x]=find(f[x]);
}
int uni(int a,int b){
	int t1,t2;
	t1=find(a);
	t2=find(b);
	if(t1!=t2){
		f[t2]=t1;
		return 1;
	}
	return 0;
}
int main(){
	ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
	int n;
	cin>>n;
	int cnt=0;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			int t;
			cin>>t;
			b[i][j]=t;
			if(i!=j){
				cnt++; 
				a[cnt].x=i;
				a[cnt].y=j;
				a[cnt].m=t;
			}
		}
	}
	bool flag=0;
	sort(a+1,a+1+cnt,cmp);
	for(int i=1;i<=n;i++) f[i]=i;
	int sum=0,ans=0;
	vector<int> res;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			if(i!=j) c[i][j]=INF;
		}
	}//除了对角线的都设成INF 
	for(int i=1;i<=cnt;i++){
		if(uni(a[i].x,a[i].y)){
			sum++;
			res.push_back(a[i].m);
			c[a[i].x][a[i].y]=a[i].m;
			c[a[i].y][a[i].x]=a[i].m;
		}
		if(sum==n-1) break;
	}
	for(int k=1;k<=n;k++)
		for(int i=1;i<=n;i++)
			for(int j=1;j<=n;j++)
				c[i][j]=min(c[i][j],c[i][k]+c[k][j]);
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			if(c[i][j]!=b[i][j]) flag=1;
		}
	}
	if(flag) cout<<"No"<<endl;
	else{
		cout<<"Yes"<<endl;
		sort(res.begin(),res.end());
		for(auto i:res)cout<<i<<endl;
	}
    return 0;
}