蓝桥杯算法提高VIP-3000米排名预测

一、题目描述

3000米长跑时，围观党们兴高采烈地预测着 最后的排名。因为他们来自不同的班，对所有运动员不一定都了解，于是他们分别对自己了解的一些运动员的实力作出了评估，即对部分运动员做了相对排名的预 测，并且告诉了可怜留守的班长。因为无聊，于是他们就组团去打Dota去了。比赛结束后他们向班长询问最后的排名，但班长不记得了，只记得他们中哪些人的 预测是正确的，哪些人的预测是错误的。他们想知道比赛的排名可能是什么。

输入
第一行两个整数n， m，n为运动员数量，m为围观党数量。运动员编号从0到n-1。
接下来m行，每行为一个围观党的相对排名预测。每行第一个数c表示他预测的人数，后面跟着c个0~n-1的不同的数，表示他预测的运动员相对排名，最后还有一个数，0表示这个预测是错误的，1表示是正确的。

数据规模和约定
1< =n< =10, 2< =c< =n, 1< =m< =10，保证数据合法，且答案中排名可能数不超过20000。对于一个排名序列，一个预测是正确的，当且仅当预测的排名的相对顺序 是排名序列的一个子序列。一个预测是错误的，当且仅当这个预测不正确。

输出
第一行一个数k为有多少种排名的可能。
下面k行，每行一个0~n-1的排列，为某一个可能的排名，相邻的数间用空格隔开。所有排名按字典序依次输出。
样例输入

3 2 
2 0 1 1 
2 1 2 0 

样例输出

2
0 2 1 
2 0 1 

二、代码与解析

  采用深度优先+剪枝的算法。

# 判断当前序列temp是否满足m个人的描述
def judge(temp):
    if len(temp) == 0:
        return True
    b = True
    for i in range(m):
        if ls[i][-1]==1:
            if not judge2(temp, ls[i][1:-1]):
                b = False
                break
        else:
            if judge3(temp, ls[i][1:-1]):
                b = False
                break
    return b

# 若围观者描述正确，判断temp是否满足
def judge2(temp, m):
    tempindex = 0
    mindex = 0
    while tempindex != len(temp) and mindex != len(m):
        if m[mindex] == temp[tempindex]:
            mindex += 1
            tempindex += 1
        else:
            tempindex += 1
    if mindex == len(m):
        return True
    else:
        while mindex != len(m):
            if m[mindex] in temp:
                return False
            mindex += 1
        return True

# 若围观者描述错误，判断temp是否满足
def judge3(temp, m):
    tempindex = 0
    mindex = 0
    while tempindex != len(temp) and mindex != len(m):
        if m[mindex] == temp[tempindex]:
            mindex += 1
            tempindex += 1
        else:
            tempindex += 1
    if mindex == len(m):
        return True
    else:
        return False

# 生成排列,深度优先用递归
def dfs(temp, nextindex):
    global num
    # 得到结果，记录
    if len(temp)== n:
        ans.append(temp[:])
        num += 1
    elif nextindex< len(container):
        # 未得到结果，向深度前进
        temp.append(container[nextindex])
        # 剪枝
        if judge(temp):
            # 前进
            po = container.pop(nextindex)
            dfs(temp, 0)
            container.insert(nextindex, po)
        # 不前进，生成下一个结点
        temp.pop()
        dfs(temp, nextindex+1)

# n, m分别保存运动员和旁观者数目，ls保存旁观者预测
n, m = list(map(int, input().strip().split()))
ls = list()
for i in range(m):
    ls.append(list(map(int, input().strip().split())))
# ans，num保存可能的结果和结果数，container保存未考虑的运动员。
ans = list()
num = 0
container = [i for i in range(n)]

dfs(list(), 0)
print(num)
for i in range(num):
    print(" ".join(list(map(str, ans[i]))))