2.10 检查排列是否合法
给定正整数N和长为N的,由N个正整数组成的向量P,设计一个算法检查P是否是1~N的一个排列
解答思路
解答1:
设置一个辅助数组Q,该数组的索引为0~N-1,初值全为0。遍历向量P,每当出现某个数字,就将数组中索引对应处(索引0对应数字1,索引1对应数字2,以此类推)存储的数值+1. 若在遍历P的过程中出现超出索引范围的数字,则返回False,表示该向量P不是1~N的一个排列。若成功遍历P,则遍历辅助数组Q,若其中任意一个元素不为1,则返回False。其他情况返回True。算法遍历两个长度为N的数组(其实更准确地说是三个,在python中生成数组Q时也要遍历一次),借助一个长度为N的数组,时间复杂度与空间复杂度均为O(N)
解答2:
若P为随机生成,考虑到整数值域之大,大部分不符合要求的向量P都是由于其中的元素大于N。因此,算法首先扫描一次向量P,将其中元素与N比较,若出现大于者,则立刻返回False。若第一遍扫描未发现大于N的元素,则检验P之中是否有重复元素。做法是选定第一个元素,将其与剩下的N-1的元素比较;若都不相等,则选定第二个元素,将其与剩下的N-2个元素比较……以此类推。若最终未发现重复元素,则可以证明向量P的确是1~N的一个排列。算法借助有限个变量,空间复杂度为O(N),但是需要将选定元素与其余成员比较,时间复杂度为O(N^2)
代码实现
# method_1
def CheckArr(P):
Q = []
N = len(P)
for i in range(N):
Q.append(0)
for i in P:
Q[i-1] += 1 # 由于list的索引从0开始,而合法排列中的最小数字是1,因此此处的索引是i-1而不是i
for i in range(N):
if(Q[i-1] != 1):
return False
return True
# method_2
def CheckArr2(P):
N = len(P)
for i in P:
if(i > N):
return False
for i in range(N):
for j in range(i+1, N): # 注意此处是i+1,否则元素跟自己比较必定返回False
if(P[i] == P[j]):
return False
return True
test1 = [1, 2, 3, 4]
test2 = [2, 2, 3, 4] # 3234再来一次
CheckArr(test1) # 返回True
CheckArr(test1) # 返回False
CheckArr2(test1) # 返回True
CheckArr2(test2) # 返回False