最小生成树+dfs

题目链接：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;
}