题意是给你一个dist[i][j]的邻接矩阵,判断这是不是一棵树。
想法是先假设这就是棵树,用最小生成树直接算出应有的n-1条边,然后暴力求出在只有这n-1条边的情况下的dist和原数组比较
当然前面还要预处理排除一堆不合法答案
hzwer:为什么要做最小生成树呢?因为首先距离当前点x最近的点y肯定是有边直接连接的,因为(反证法)假设有z使得xz和yz分别连接,则dis[x][z]+dis[y][z]<=dis[x][y],所以应该有dis[x][z]<dis[x][y],与已知条件dist[x][y]最小不符
所以我们要优先考虑边权小的边,所以直接最小生成树
这题n=2000就是400w的边,再加点处理也有200w边,明显稠密图,应该用Prim,居然Kruskal能过……服了
贴代码……
以下Kruskal版
#include<cstdio>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#include<queue>
#include<deque>
#include<set>
#include<map>
#include<ctime>
#define LL long long
#define inf 2147483647
#define pa pair<int,int>
#define N 2100
using namespace std;
struct bian{
int x,y,z;
}b[2000010];
bool operator < (const bian &a,const bian &b)
{
return a.z<b.z;
}
struct edge{
int to,next,v;
}e[10*N];int head[N];
LL n,cnt,tot;
LL a[N][N];
int fa[N];
int top,zhan[N];bool vis[N];
LL dist[N][N];
inline int getfa(int x)
{return fa[x]==x?x:fa[x]=getfa(fa[x]);}
inline void ins(int u,int v,int w)
{
e[++cnt].to=v;
e[cnt].next=head[u];
e[cnt].v=w;
head[u]=cnt;
}
inline void insert(int u,int v,int w)
{
ins(u,v,w);
ins(v,u,w);
}
inline LL read()
{
LL x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
inline void init()
{
n=read();
for (int i=1;i<=n;i++)
for (int j=1;j<=n;j++)
a[i][j]=read();
}
inline bool pre_judge()
{
for (int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
{
if (i==j&&a[i][i]!=0)return 0;
if (i!=j&&a[i][j]==0)return 0;
if (a[i][j]!=a[j][i])return 0;
}
return 1;
}
inline void Kruskal()
{
for (int i=1;i<=n;i++)fa[i]=i;
for (int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
if (i<j)
{
b[++tot].x=i;
b[tot].y=j;
b[tot].z=a[i][j];
}
sort(b+1,b+tot+1);
int piece=n;
for (int i=1;i<=tot;i++)
{
int fx=getfa(b[i].x);
int fy=getfa(b[i].y);
if (fx==fy)continue;
piece--;
fa[fx]=fy;
insert(b[i].x,b[i].y,b[i].z);
if (piece==1)return;
}
}
inline void dfs(int cur)
{
for (int i=head[cur];i;i=e[i].next)
{
if (vis[e[i].to])continue;
for (int j=1;j<=top;j++)
{
dist[e[i].to][zhan[j]]=dist[zhan[j]][e[i].to]=dist[zhan[j]][cur]+e[i].v;
}
zhan[++top]=e[i].to;
vis[e[i].to]=1;
dfs(e[i].to);
}
}
int main()
{
init();
if (!pre_judge())
{
printf("NO");
return 0;
}
Kruskal();
zhan[1]=1;top=1;vis[1]=1;
dfs(1);
for (int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
if(a[i][j]!=dist[i][j])
{
printf("NO");
return 0;
}
printf("YES");
return 0;
}