今天做多源最短路,遇到下面问题:
图片:
只拿了90分(满分100)
我的代码如下:
#include<bits/stdc++.h>
using namespace std;
int n,m,d[105][105],p;
long long anss=0x7fffffff;
bool bo[105],vis[105];
void dfs(int x,int m,long long ss)
{
if(m>=p)
{
anss=min(anss,ss+d[x][n]);
return;
}
else
{
for(int i=1;i<=n;i++)
{
if(!vis[i]&&bo[i])
{
vis[i]=true;
dfs(i,m+1,ss+d[x][i]);
vis[i]=false;
}
}
}
}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
{
scanf("%d",&d[i][j]);
}
}
for(int k=1;k<=n;k++)
{
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
{
if(d[i][k]!=0&&d[k][j]!=0)
d[i][j]=min(d[i][j],d[i][k]+d[k][j]);
}
}
}
scanf("%d",&p);
for(int i=1;i<=p;i++)
{
int x;
scanf("%d",&x);
bo[x]=true;
}
vis[1]=true;
dfs(1,1,0);
printf("%lld",anss);
}