PAT甲级1150

PAT甲级1150

题目大意：给出一个图以及几组路径判断这些路径是哪一类路径，并且最后一行输出最短路径以及是第几组路径。路径种类：1、构成一个环且所有节点（除开始结尾的节点）只访问一次：TS simple cycle；2、构成环但是除首尾节点访问超过1次或者首尾节点访问超过2次：TS cycle；3、未能构成一个环或者虽构成环但是不是图中每个节点都访问到：Not a TS cycle。
本以为是个图论的最短路径啥的，结果只是个简单模拟分类。首先首尾节点必须相同否则一定是第三类（未构成环）；如果碰到路径途中有两个节点之间没有边则一定是第三类，且路径长度为NA；设置一个visit数组记录每个节点访问的次数以及cnt记录总共访问了多少个顶点（visit为0的时候加1即可），若是cnt<n则也是第三类。接下来判断是否是第二类：遍历visit数组，如果首尾节点访问次数大于2或者非首尾节点访问次数大于1。如果确定是第二类则用flag标记，否则会重复输出。剩下的只要flag是true就是第一类。至于最短路径以及下标额外用两个变量index和mindist记录，初始化mindist为INF，碰到比它小的就更新下标和长度即可。

#include <iostream>
#include <vector>
#include <algorithm>
#define MAXN 202
#define INF 99999999
using namespace std;

int n,m;
int graph[MAXN][MAXN];
int main(){
	int city1,city2,dist,k,mindist=INF,index;
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
		fill(graph[i]+1,graph[i]+n+1,INF);
	for(int i=0;i<m;i++){
		scanf("%d%d%d",&city1,&city2,&dist);
		graph[city1][city2]=graph[city2][city1]=dist;
	}
	scanf("%d",&k);
	for(int i=1;i<=k;i++){
		bool flag=true;
		int visit[n+1]={0};
		int sum,cnt=0,start,now,next,distance=0;
		scanf("%d",&sum);
		scanf("%d",&start);
		visit[start]++,cnt++;
		now=start;
		for(int j=1;j<sum;j++){
			scanf("%d",&next);
			if(flag){
				if(graph[now][next]==INF){	//两个城市之间没有路，之后的数据都不用再考虑了
					flag=false;
				}else{
					if(visit[next]==0) cnt++;
					visit[next]++;
					distance+=graph[now][next];
					now=next;
				}
			}
		}
		//结束循环后next即为终点
		if(flag==false){
			printf("Path %d: NA (Not a TS cycle)\n",i);
			continue;
		}else if(cnt!=n||start!=next){	//未访问所有顶点或者起终点不一样
			printf("Path %d: %d (Not a TS cycle)\n",i,distance);
			continue;
		}
		flag=true;
		if(mindist>distance){
			mindist=distance;
			index=i;
		}
		for(int j=1;j<=n;j++){
			if(j!=start&&j!=next&&visit[j]>1||visit[j]>2){	//除了起终点以外有别的点访问超过1次或者有的顶点访问超过2次
				flag=false;
				printf("Path %d: %d (TS cycle)\n",i,distance);
				break;
			}
		}
		if(flag) printf("Path %d: %d (TS simple cycle)\n",i,distance);
	}
	printf("Shortest Dist(%d) = %d",index,mindist);
	system("pause");
	return 0;
}