一道值得分享(interesting)的题目

ps：前几天，博主不知道通过什么方式拿到了一个题目。

题目梗概

给你一个$n*n$的矩阵，要求判断其是否满足一下要求：
1对于任何$i$满足.$a[i][i]=0$
2.对于任何$i,j$满足$a[i][j]=a[j][i]$
3.对于任何$i,j,k$满足$a[i][j]<=max(a[i][k],a[j][k])$
输出$YES$或$NO$
$n<=2000$

解题思路

这个题目，描述清新易懂，一看就是水题很可做。
条件1，2都很好判断，注意考虑3。

拿到手应该有个十分暴力的想法，就是枚举当前的最大权值,标记所有小于这个权值的点(如果小于这个值标记1否则为0)，考虑刚好等于当前权值的一些点，条件3就是把$i,j$两行拿来对比，对于每一列必须有一个$>=a[i][j]$，所以把标记看成二进制数，并起来等于0就满足条件，然后用bitset乱搞(bitset大法好啊)，前面用$a[i][j]$建图就可做到$O(N^2*a)$，$a$为bitset操作系数。

这样的做法的确回过(如果手写bitset的话)，但是如果题目仅仅只是这一种操作就不叫值得分享(interesting)。

我想细心的读者已经发现，这个题目的条件很有意思（如果没有发现的人请再读一遍题面）。

这TM不就是一个关于$n$个点的邻接矩阵（原来出题人打着矩阵的幌子搞事情。出题人:这不也是矩阵吗！= =）。

也许发现这个现象之后也没有立刻得到清新的思路，但是我们有个侧重方向，条件3要求$a[i][j]<=max(a[i][k],a[j][k])$，可以从比较小的边先开始判断，所以想到最小生成树。
假设我们已经得到一个这样的最小生成树。

考虑节点$i$和其他节点是否满足条件3，先来按深度分析。

第一层，显然满足。
第二层，因为是最小生成树，显然有$a[i][j]>=D(i,j)$,$D(i,j)$表示$i到j$路径上的最大边的边权。因为还要满足条件3，所以$a[i][j]==D(i,j)$。
第三层，显然对于$i,j$只需要考虑他们路径上在第二层的k的情况
$a[i][j]>=max(a[j][k]，a[i][k])$，由第二层结论得到
$a[i][j]>=max(a[j][k]，D(i,k))=D(i,j)$,因为有最小生成树的结论，所以$a[i][j]==D(i,j)$。
第四层，跟上一层同样考虑和第三层的关系。
所以最后得出结论，满足条件3等价于满足$a[i][j]==D(i,j)$。

思路瞬间清晰，代码就不难实现了。

#include<cstdio>
#include<cstring>
using namespace std;
const int maxn=2005,oo=2147483647,maxm=4005;
inline int _read(){
    int num=0;char ch=getchar();
    while (ch<'0'||ch>'9') ch=getchar();
    while (ch>='0'&&ch<='9') num=num*10+ch-48,ch=getchar();
    return num;
}
int n,a[maxn][maxn],dis[maxn],fa[maxn],b[maxn][maxn];
int nxt[maxm],lnk[maxn],son[maxm],w[maxm],tot;
bool vis[maxn];
void add(int x,int y,int z){
    nxt[++tot]=lnk[x];lnk[x]=tot;son[tot]=y;w[tot]=z;
}
int max(int x,int y){if (x>y) return x;return y;}
void DFS(int *b,int x,int f){
    for (int j=lnk[x];j;j=nxt[j])
    if (son[j]!=f){
        b[son[j]]=max(b[x],w[j]);
        DFS(b,son[j],x);
    }
}
int main(){
    freopen("grail.in","r",stdin);
    freopen("grail.out","w",stdout);
    n=_read();
    for (int i=1;i<=n;i++)
    for (int j=1;j<=n;j++) a[i][j]=_read();
    for (int i=1;i<=n;i++)
    for (int j=1;j<=i;j++){
        if (i==j&&a[i][i]!=0) return printf("NO"),0;
        if (a[i][j]!=a[j][i]) return printf("NO"),0;
    }
    for (int i=1;i<=n;i++) dis[i]=a[i][1],fa[i]=1;
    vis[1]=1;
    while(1){
        int mi=oo,j;
        for (int i=1;i<=n;i++)
        if (!vis[i]&&dis[i]<mi) mi=dis[i],j=i;
        if (mi==oo) break;
        add(fa[j],j,mi);add(j,fa[j],mi);
        vis[j]=1;
        for (int i=1;i<=n;i++)
        if (!vis[i]&&a[i][j]<dis[i]) dis[i]=a[i][j],fa[i]=j;
    }
    for (int i=1;i<=n;i++){
        DFS(b[i],i,0);
        for (int j=1;j<=n;j++)
        if (a[i][j]!=b[i][j]) return printf("NO"),0;
    }
    return printf("YES"),0;
}