【PTA】L3-015 球队“食物链” (30 分)（DFS+剪枝）

最新推荐文章于 2021-09-26 09:43:10 发布

海马有力量

最新推荐文章于 2021-09-26 09:43:10 发布

阅读量395

点赞数

分类专栏： DFS pta 数据结构

本文链接：https://blog.csdn.net/qq_20087731/article/details/115754441

版权

数据结构同时被 3 个专栏收录

52 篇文章 3 订阅

订阅专栏

pta

38 篇文章 5 订阅

订阅专栏

DFS

5 篇文章 0 订阅

订阅专栏

L3-015 球队“食物链” (30 分)

在这里插入图片描述
输入样例1：

5
-LWDW
W-LDW
WW-LW
DWW-W
DDLW-

输出样例1：

1 3 5 4 2

输入样例2：

5
-WDDW
D-DWL
DD-DW
DDW-D
DDDD-

输出样例2：

No Solution

dfs遍历，要注意的是两两相互对局，给出的样例图并不是对称的。写到有一个测试点超时时，就感觉突破口是食物链最后的点连接着第一个点。

#include<bits/stdc++.h>
using namespace std;
#define mk make_pair
const int N=30;
char a[N][N];
int n,v1[N],v2[N],flag;
vector<int>ans;
void dfs(int pos){
	if(flag) return;
	int f=0;
	for(int i=0;i<n;i++){
		if(v1[i]==0&&(a[i][0]=='W'||a[0][i]=='L')){
			f=1;break;
		}
	}
	if(!f) return;
	ans.pb(pos);
	if(ans.size()==n){
		if(a[pos][0]=='W'||a[0][pos]=='L'){
			for(int i=0;i<n;i++){
				if(i) printf(" ");
				printf("%d",ans[i]+1);
			}
			flag=1;	
		}
		ans.pop_back();
		return;
	}
	v1[pos]=1;
	for(int i=0;i<n;i++){
		if(v1[i]==0&&(a[pos][i]=='W'||a[i][pos]=='L')){
			dfs(i);			
		}
	}
	ans.pop_back();
	v1[pos]=0;
}
int main() 
{
	scanf("%d",&n);
	for(int i=0;i<n;i++){
		scanf("%s",&a[i]);
	}
	dfs(0);
	if(flag==0) puts("No Solution");
 	return 0;
}