题目
https://pintia.cn/problem-sets/994805342720868352/problems/1038430013544464384
题意
判断给出的路径是否是符合要求的旅行商环路
题解
将所给的路径关系放入二维数组 f 中,在判断所给路径时,通过f数组判断是否为NA,通过判断set的容量、len与n的关系进行相应的输出。如果是旅行商环路,通过min和pos两个变量存储最短旅行商环路长度及其位置,并在最后输出
AC代码
#include<bits/stdc++.h>
using namespace std;
int main()
{
int n,m,k,f[205][205];
scanf("%d %d",&n,&m);
memset(f,0,sizeof(f));
while(m--)
{
int x,y,z;
scanf("%d %d %d",&x,&y,&z);
f[x][y]=f[y][x]=z;
}
scanf("%d",&k);
int min=INT_MAX,pos=-1;
for(int t=1;t<=k;t++)
{
int len,flag=1,sum=0;
cin>>len;
set<int> s;
int a[len];
for(int i=0;i<len;i++)
{
cin>>a[i];
s.insert(a[i]);
}
for(int i=0;i<len-1;i++)
{
if(f[a[i]][a[i+1]]==0) flag=0;
sum+=f[a[i]][a[i+1]];
}
if(flag==0) printf("Path %d: NA (Not a TS cycle)\n",t);
else if(a[0]!=a[len-1]||s.size()!=n) printf("Path %d: %d (Not a TS cycle)\n",t,sum);
else if(len-1!=n)
{
printf("Path %d: %d (TS cycle)\n",t,sum);
if(sum<min)
{
min=sum;
pos=t;
}
}
else
{
printf("Path %d: %d (TS simple cycle)\n",t,sum);
if(sum<min)
{
min=sum;
pos=t;
}
}
}
printf("Shortest Dist(%d) = %d",pos,min);
}
参考
PAT 1150 Travelling Salesman Problem(25 分)- 甲级
注意
n和len不要弄混,一个是总的结点数,一个是所给路径的节点数