L3-015. 球队“食物链”
某国的足球联赛中有N支参赛球队,编号从1至N。联赛采用主客场双循环赛制,参赛球队两两之间在双方主场各赛一场。
联赛战罢,结果已经尘埃落定。此时,联赛主席突发奇想,希望从中找出一条包含所有球队的“食物链”,来说明联赛的精彩程度。“食物链”为一个1至N的排列{ T1 T2 ... TN },满足:球队T1战胜过球队T2,球队T2战胜过球队T3,……,球队T(N-1)战胜过球队TN,球队TN战胜过球队T1。
现在主席请你从联赛结果中找出“食物链”。若存在多条“食物链”,请找出字典序最小的。
注:排列{ a1 a2 ...aN }在字典序上小于排列{ b1 b2 ... bN },当且仅当存在整数K(1 <= K <= N),满足:aK < bK且对于任意小于K的正整数i,ai=bi。
输入格式:
输入第一行给出一个整数N(2 <= N <= 20),为参赛球队数。随后N行,每行N个字符,给出了NxN的联赛结果表,其中第i行第j列的字符为球队i在主场对阵球队j的比赛结果:“W”表示球队i战胜球队j,“L”表示球队i负于球队j,“D”表示两队打平,“-”表示无效(当i=j时)。输入中无多余空格。
输出格式:
按题目要求找到“食物链”T1 T2 ... TN,将这N个数依次输出在一行上,数字间以1个空格分隔,行的首尾不得有多余空格。若不存在“食物链”,输出“No Solution”。
输入样例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有点小问题(脑袋懵圈了)只拿了16分。现在又重新研究一下。
题目可以抽象成一个图论的问题,如果i号节点赢过j号节点,就可以在i和j之间加一条有向边。然后从1号开始搜这个图,如果能找到回路,那么字典序最小一定是从1开始(因为既然都是回路了,所以按字典序排1号一定是第一个)。如果dfs没有回路就输出No Solution。注意dfs的时候就从字典序开始搜,这样一旦找到结果就是最小的字典序,此时直接flag = true,return即可(剪枝)。如果只写这个剪枝只能得22分,还有一个地方能剪枝:在dfs的过程中,如果剩下的点没有和1节点相连的,就直接返回。
另外需要注意一个地方,如果你想用邻接表建图,如果是在读输入的时候就进行push_back边,那么dfs之前一定要排序一下,因为不排序邻接表可能不是按字典序排的。详细可以看我代码注释。
#include <cstdio>
#include <vector>
#include <cstring>
#include <algorithm>
#define MAX 20 + 10
using namespace std;
vector<int> MGraph[MAX];
vector<int> vec;
int visit[MAX];
int n;
int MAP[MAX][MAX];
bool flag = false;
void dfs( int cur, int start, int depth ) {
if( flag ) return;
if( depth == n ) {
int temp = false;
for( int i = 0; i < MGraph[cur].size(); i++ ) {
int v = MGraph[cur][i];
if( v == start ) {
temp = true;
break;
}
}
if( temp ) {
flag = true;
for( int i = 0; i < vec.size(); i++ ) {
if( i == 0 ) printf( "%d", vec[i] + 1 );
else printf( " %d", vec[i] + 1 );
}
printf( "\n" );
}
return;
}
int x;
for( x = 0; x < n; x++ ) { // 如果剩下的点没有和起点相连的,就直接返回
if( !visit[x] && MAP[x][0] == 1 ) break;
}
if( x == n ) return;
for( int i = 0; i < MGraph[cur].size(); i++ ) {
int v = MGraph[cur][i];
if( !visit[v] ) {
visit[v] = 1;
vec.push_back( v );
dfs( v, start, depth + 1 );
vec.pop_back();
visit[v] = 0;
}
}
}
int main() {
char str[MAX];
scanf( "%d", &n );
getchar();
for( int i = 0; i < n; i++ ) {
scanf( "%s", str );
for( int j = 0; j < strlen( str ); j++ ) {
if( str[j] == 'W' ) {
MAP[i][j] = 1;
MGraph[i].push_back( j );
}
if( str[j] == 'L' ) {
MAP[j][i] = 1;
MGraph[j].push_back( i );
}
}
}
// 注意如果是读输入的时候就建图,这样字典序的顺序就不一样了!
for( int i=0;i<n;i++) {
sort(MGraph[i].begin(),MGraph[i].end());
//MGraph[i].erase(unique(MGraph[i].begin(),MGraph[i].end()),MGraph[i].end());
}
/*
for( int i = 0; i < n; i++ )
for( int j = 0; j < n; j++ ) {
if( MAP[i][j] ) MGraph[i].push_back( j );
}
*/
vec.push_back( 0 );
visit[0] = 1;
dfs( 0, 0, 1 );
if( !flag ) printf( "No Solution\n" );
return 0;
}