C - Anna, Svyatoslav and Maps floyd

 

题意:给n*n的01矩阵,1i与j联通,0不联通,给m个点,表示依次走m个点,要求这些点的最短子序列,使最短子序列里的点走完的最短路与走完m个点的相同。

思路:先floyd预处理两两点之间的最短距离,预处理时需要注意i到i点距离为0,之后将第一个点放入答案,枚举答案的最后一个顶点与m中第二个顶点开始的点的距离,如果距离相同,可以不选择m中的这个点,如果距离变小,答案中需要加上这个点的前一个点。

#include<bits/stdc++.h>
using namespace std;
const int INF=0x3f3f3f3f;
const int maxm=1e6+10;
char mp[200][200];
int e[200][200];
int l[maxm];
int ans[maxm];
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1; i<=n; i++)
        scanf("%s",mp[i]+1);
    for(int i=1; i<=n; i++)
        for(int j=1; j<=n; j++)
        {
            if(i==j) e[i][j]=0;
            else e[i][j]=INF;
        }
    for(int i=1; i<=n; i++)
        for(int j=1; j<=n; j++)
        {
            if(mp[i][j]=='1')
                e[i][j]=1;
        }
    for(int k=1; k<=n; k++)
        for(int i=1; i<=n; i++)
        {
            for(int j=1; j<=n; j++)
            {
                if(e[i][k]<INF&&e[k][j]<INF&&e[i][j]>e[i][k]+e[k][j])
                    e[i][j]=e[i][k]+e[k][j];
            }
        }
    int m;
    scanf("%d",&m);
    for(int i=1; i<=m; i++)
        scanf("%d",&l[i]);
    int cnt=0;
    ans[++cnt]=l[1];
    int dis=0;

    for(int i=2; i<=m; i++)
    {
        dis+=e[l[i-1]][l[i]];
        if(e[ans[cnt]][l[i]]<dis)
        {
            ans[++cnt]=l[i-1];
            dis=e[ans[cnt]][l[i]];
        }
    }
    ans[++cnt]=l[m];
    printf("%d\n",cnt);
    for(int i=1; i<=cnt; i++)
        printf("%d ",ans[i]);
}

 

上一篇:【关系闭包】传递闭包 Warshall Floyd


下一篇:Codeforces Gym 101252D&&floyd判圈算法学习笔记