题目链接:CodeForces - 472D
题目大意:告诉你一棵树有n个节点,然后以矩阵的形式告诉你每两个节点间的距离,问是否能构成一颗树。
首先,我们可以排除掉一些错误答案,比如arr[i][i]!=0,arr[i][j]!=arr[j][i],arr[i][j]=0(i!=j)这几种我们可以直接输出NO
求最小生成树,然后dfs求两点之间的最短距离,是否符合已知情况
代码 :
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cstring>
#include <string>
#include <algorithm>
#include <set>
#include <map>
#include <stack>
#include <vector>
#include <queue>
#define ri(n) scanf("%d",&n)
#define oi(n) printf("%d\n",n)
#define rl(n) scanf("%lld",&n)
#define ol(n) printf("%lld\n",n)
#define rep(i,l,r) for(i=l;i<=r;i++)
#define rep1(i,l,r) for(i=l;i<r;i++)
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int epg=10-8;
ll dis[100000+10];
ll a[2000+10][2000+10];
int p[2000+10];
ll dist[2000+10];
int Find(int x)
{
if(p[x]==x)
return x;
return p[x]=Find(p[x]);
}
struct node
{
int u,v;
ll d;
node(int uu=0,int vv=0,ll dd=0):u(uu),v(vv),d(dd) {}
} e[2000*2000+10];
int sum,flag;
int n;
vector<node>G[2000+10];
void dfs(int u,int s)//搜索u结点到其他结点的距离
{
for(int i=0; i<G[u].size(); i++)
{
int v=G[u][i].v;
ll d=G[u][i].d;
if(v==s)
continue;
dist[v]=dist[u]+d;
dfs(v,u);
}
}
bool cmp(node x,node y)
{
return x.d<y.d;
}
void kusurl()
{
ll ans=0;
int flag2=0;
for(int i=0; i<2000+10; i++)
{
p[i]=i;
G[i].clear();
}
int cnt=0;
for(int i=0; i<sum; i++)
{
int fx=Find(e[i].u);
int fy=Find(e[i].v);
if(fx!=fy)
{
ans+=e[i].d;
p[fx]=fy;
cnt++;
G[e[i].u].push_back(node(e[i].u,e[i].v,e[i].d));
G[e[i].v].push_back(node(e[i].v,e[i].u,e[i].d));
}
//if(cnt==sum-1)
//cout<<"ans="<<ans<<endl;
//cout<<"cnt="<<cnt<<" sum="<<sum<<endl;
if(cnt==n-1)
{
//flag=1;
flag2=1;
break;
}
}
//cout<<"flag2="<<flag2<<endl;
if(flag2==0)
{
flag=0;
return;
}
for(int i=1; i<=n; i++)
{
dist[i]=0;
dfs(i,0);
//dfs(i,-1);
for(int j=i+1; j<=n; j++)
{
if(dist[j]!=a[i][j])
{
flag=0;
break;
}
}
if(flag==0)
break;
}
}
int main()
{
//int n;
while(scanf("%d",&n)==1)
{
flag=1;
sum=0;
for(int i=1; i<=n; i++)
for(int j=1; j<=n; j++)
{
scanf("%lld",&a[i][j]);
if(i==j&&a[i][j]!=0)
{
flag=0;
}
}
for(int i=1; i<=n; i++)
for(int j=i+1; j<=n; j++)
{
if(a[i][j]!=0)
{
//sum++;
//e[i][j]=e[j][i]=a[i][j];
e[sum].u=i;
e[sum].v=j;
e[sum++].d=a[i][j];
}
}
for(int i=1; i<=n; i++)
{
for(int j=i+1; j<=n; j++)
{
if((a[i][j]!=0&&a[j][i]!=0&&a[i][j]!=a[j][i])||a[i][j]==0)
//if(a[i][j]!=a[j][i]||a[i][j]==0)
{
flag=0;
}
}
}
//for(int i=0; i<sum; i++)
//cout<<"*"<<e[i].u<<" "<<e[i].v<<" "<<e[i].d<<endl;
if(flag==0)
{
printf("NO\n");
continue;
}
else
{
sort(e,e+sum,cmp);
kusurl();
if(flag||(n==1&&a[1][1]==0))
printf("YES\n");
else
printf("NO\n");
}
}
return 0;
}