1150 Travelling Salesman Problem (25 分)

写了半天终于对了，还是细节问题QWQ

#include<iostream>
#include<string>
#include<stdio.h>
#include<string.h>
#include<vector>
#include<map>
#include<set>
#include<unordered_map>
#include<queue>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long LL;
const int maxn=210;
vector<int> G[maxn];
bool istcycle[maxn],iscycle[maxn];
int weight[maxn][maxn];
int n,m;
int mindis=9999999;
int q;

void judge(int a[],int len,int id){
    memset(istcycle,false,sizeof(istcycle));
    memset(iscycle,false,sizeof(iscycle));
    int totaldis=0;
    int sflag=0,cflag=0;
    for(int i=0;i<len-1;i++){
        int u=a[i],v=a[i+1];
        vector<int>::iterator it;
        for(it=G[u].begin();it!=G[u].end();it++) if(*it==v) break;
        if(it==G[u].end()){
            printf("Path %d: NA (Not a TS cycle)\n",id);
            return;
        }
        else{
            if(istcycle[v]==false){ //判断是否是简单回路
                istcycle[v]=true;
            }
            else{
                sflag=1;
            }
            iscycle[v]=true;
            totaldis+=weight[u][v];
        }
    }
    for(int i=1;i<=n;i++){ //判断所有节点是否都覆盖
        if(iscycle[i]==false){
            cflag=1;
            break;
        }
    }
    if(a[0]!=a[len-1]) {//首尾不一致一定不是环
        sflag=1,cflag=1;
    }
    if(totaldis<mindis&&cflag==0){ //是回路才能更新路径
        mindis=totaldis;
        q=id;
    }
    if(sflag==0&&cflag==0) printf("Path %d: %d (TS simple cycle)\n",id,totaldis);
    else if(cflag==0) printf("Path %d: %d (TS cycle)\n",id,totaldis);
    else printf("Path %d: %d (Not a TS cycle)\n",id,totaldis);
}

int main()
{
    int a[maxn];
    scanf("%d%d",&n,&m);
    int u,v,w;
    for(int i=0;i<m;i++){
        scanf("%d%d%d",&u,&v,&w);
        G[v].push_back(u);
        G[u].push_back(v);
        weight[u][v]=weight[v][u]=w;
    }
    int query,k;
    scanf("%d",&query);
    for(int i=0;i<query;i++){
        scanf("%d",&k);
        for(int j=0;j<k;j++) {
            scanf("%d",&a[j]);
        }
        judge(a,k,i+1);
    }
    printf("Shortest Dist(%d) = %d\n",q,mindis);
}