【PAT】1150 Travelling Salesman Problem (25 分)

fftx_00

已于 2022-02-10 20:46:20 修改

阅读量325

点赞数

分类专栏： PAT_advanced2 文章标签： C++ 图算法旅行商问题最短路径路径搜索

于 2022-02-08 20:25:06 首次发布

本文链接：https://blog.csdn.net/fftx_00/article/details/122830670

版权

PAT_advanced2 专栏收录该内容

35 篇文章 0 订阅

订阅专栏

#include <bits/stdc++.h>
using namespace std;
#define maxn 210
int n,m,k,tk;
struct node
{
    int v,dist;
};

vector<node> G[maxn];//[1,n]
int kPath[maxn];//[1,tk]
int sid=-1,ss=INT_MAX;

int vis[maxn];
void isTS(int times){
    fill(vis,vis+maxn,0);
    int ans=0,fflag=1,flag=0;
    vis[kPath[1]]=1;
    for(int i=2;i<=tk;i++){
        flag=0;
        for(int j=0;j<G[kPath[i-1]].size();j++){
            if(kPath[i]==G[kPath[i-1]][j].v){
                flag=1;
                ans+=G[kPath[i-1]][j].dist;
                vis[kPath[i]]++;
                break;
            }
        }
        if(flag==0){
            fflag=0;break;
        }
    }

    if(fflag==0){//路径非法->NA
        ans=-1;
    }

    for(int i=1;i<=n;i++){//每个城市都访问了
        if(vis[i]==0){
                fflag=0;
                break;
        }
    }

    if(fflag==1 && kPath[1]==kPath[tk]){//路径合法&每个城市都访问了
        if(vis[kPath[1]]==2)    printf("Path %d: %d (TS simple cycle)\n",times,ans);
        else                    printf("Path %d: %d (TS cycle)\n",times,ans);
    }else{
        if(ans==-1)             printf("Path %d: %s (Not a TS cycle)\n",times,"NA");
        else                    printf("Path %d: %d (Not a TS cycle)\n",times,ans);
    }

    if(fflag==0) ans=-1;
    if(ans!=-1 && ans<ss){
        ss=ans,sid=times;
    }
}

int main(){
    scanf("%d %d",&n,&m);

    int c1,c2,cd;
    for(int i=0;i<m;i++){
        scanf("%d %d %d",&c1,&c2,&cd);
        G[c1].push_back(node{c2,cd});
        G[c2].push_back(node{c1,cd});
    }

    scanf("%d",&k);

    for(int i=0;i<k;i++){
        scanf("%d",&tk);
        for(int t=1;t<=tk;t++){
            scanf("%d",&kPath[t]);
        }
        isTS(i+1);
    }
    
    if(sid!=-1)     printf("Shortest Dist(%d) = %d",sid,ss);

    return 0;
}

//没啥区别,除了求最小path的方式
#include <bits/stdc++.h>
using namespace std;
#define maxn 210
int n,m,k,tk;
struct node
{
    int v,dist;
};

vector<node> G[maxn];//[1,n]
int kPath[maxn];//[1,tk]
int shortPath[maxn*10];//题中没给k范围,如果要存储注意试一下,或者用vector

int vis[maxn];
void isTS(int times){
    fill(vis,vis+maxn,0);
    int ans=0,fflag=1,flag=0;
    vis[kPath[1]]=1;
    for(int i=2;i<=tk;i++){
        flag=0;
        for(int j=0;j<G[kPath[i-1]].size();j++){
            if(kPath[i]==G[kPath[i-1]][j].v){
                flag=1;
                ans+=G[kPath[i-1]][j].dist;
                vis[kPath[i]]++;
                break;
            }
        }
        if(flag==0){
            fflag=0;break;
        }
    }

    if(fflag==0){//路径非法->NA
        ans=-1;
    }

    for(int i=1;i<=n;i++){//每个城市都访问了
        if(vis[i]==0){
                fflag=0;
                break;
        }
    }

    if(fflag==1 && kPath[1]==kPath[tk]){//路径合法&每个城市都访问了
        if(vis[kPath[1]]==2)    printf("Path %d: %d (TS simple cycle)\n",times,ans);
        else                    printf("Path %d: %d (TS cycle)\n",times,ans);
    }else{
        if(ans==-1)             printf("Path %d: %s (Not a TS cycle)\n",times,"NA");
        else                    printf("Path %d: %d (Not a TS cycle)\n",times,ans);
    }

    if(fflag==0) ans=-1;
    shortPath[times]=ans;
}

int main(){
    scanf("%d %d",&n,&m);

    int c1,c2,cd;
    for(int i=0;i<m;i++){
        scanf("%d %d %d",&c1,&c2,&cd);
        G[c1].push_back(node{c2,cd});
        G[c2].push_back(node{c1,cd});
    }

    scanf("%d",&k);

    for(int i=0;i<k;i++){
        scanf("%d",&tk);
        for(int t=1;t<=tk;t++){
            scanf("%d",&kPath[t]);
        }
        isTS(i+1);
    }
    int shortId=-1,shortS=INT_MAX;
    for(int i=1;i<=k;i++){
        if(shortPath[i]!=-1 && shortPath[i]<shortS){
            shortS=shortPath[i];
            shortId=i;
        }
    }
    if(shortId!=-1) printf("Shortest Dist(%d) = %d",shortId,shortS);

    return 0;
}