codeforces1204 C. Anna, Svyatoslav and Maps (floyd)

题目链接：http://codeforces.com/contest/1204/problem/C
题目解法：用dis(i,j)来表示i到j的最短路，若dis(i,k)+dis(k,j)==dis(i,j)，那么k可以删掉。

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int inf=1e9;
char s[105];
int N,ma[105][105];
int M,p[1000006];

int main()
{
	//输入
    scanf("%d",&N);
    for(int i=1;i<=N;i++)
    {
        scanf("%s",s+1);
        for(int j=1;j<=N;j++)
        {
            if(s[j]=='1')ma[i][j]=1;
            else ma[i][j]=inf;
        }
        ma[i][i]=0;
    }
    scanf("%d",&M);
    for(int i=1;i<=M;i++)
        scanf("%d",&p[i]);
        
        
    //floyd 求最短路
    for(int k=1;k<=N;k++)
        for(int i=1;i<=N;i++)
            for(int j=1;j<=N;j++)
                if(ma[i][k]!=inf&&ma[k][j]!=inf)
                    ma[i][j]=min(ma[i][j],ma[i][k]+ma[k][j]);

    
    //删点
    int idx=1,cnt=2,ans=0;
    while(idx+cnt<=M)
    {
        if(ma[p[idx]][p[idx+cnt]]==ma[p[idx]][p[idx+cnt-1]]+ma[p[idx+cnt-1]][p[idx+cnt]])
        {
            p[idx+cnt-1]=-1;
            ans++;
            cnt++;
        }
        else
        {
            idx=idx+cnt-1;
            cnt=2;
        }
    }
    
    
    //输出答案
    printf("%d\n",M-ans);
    for(int i=1;i<=M;i++)
        if(p[i]!=-1)printf("%d ",p[i]);
        
    return 0;
}