cf472D Design Tutorial: Inverse the Problem

题意是给你一个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; 
}