题目链接:https://vjudge.net/problem/CodeForces-1204C
转自:https://blog.csdn.net/qq_40831340/article/details/100141541
题意:有一个有向无环图,有一串序列,让你尽可能地删除更多的点,使得剩下点的序列按照最短路走仍是原序列。
思路:先跑floyd求两点距离,接下来遍历,从上一个点p出发,统计距离,如果p到这个点的距离大于最短路的距离,说明应该把当前点保留,如果删去,则最短路会变短,然后更新p点。
#include <bits/stdc++.h>
using namespace std;
const int maxn=150;
int g[maxn][maxn];
const int inf=0x3f3f3f3f;
int ans[1000005],a[1000005];
char s[150];
int main()
{
ios::sync_with_stdio(false);
int n;
cin>>n;
memset(g,inf,sizeof(g));
for(int i=1; i<=n; i++)
{
cin>>s;
for(int j=0; j<n; j++)
{
int idx=s[j]-'0';
if(idx)
{
g[i][j+1]=1;
}
if(i==j+1)
g[i][j+1]=0;
}
}
for(int k=1; k<=n; k++)
{
for(int i=1; i<=n; i++)
{
for(int j=1; j<=n; j++)
{
g[i][j]=min(g[i][k]+g[k][j],g[i][j]);
}
}
}
int k;
cin>>k;
for(int i=1; i<=k; i++)
{
cin>>a[i];
}
int cnt=0;
ans[++cnt]=a[1];
int dis=0;
for(int i=2; i<=k; i++)
{
dis+=g[a[i-1]][a[i]];
if(dis>g[ans[cnt]][a[i]])
{
ans[++cnt]=a[i-1];
dis=g[ans[cnt]][a[i]];
}
}
ans[++cnt]=a[k];
cout<<cnt<<endl;
for(int i=1; i<=cnt; i++)
{
if(i!=1)
cout<<" ";
cout<<ans[i];
}
return 0;
}