题意:给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]); }