题目描述:
思路:
这道题目乍看之下没有特别适用的算法,于是尝试用深度优先的方法直接暴力搜索。根据题干描述,最后生成序列的长度为2n-1。为了满足第三个条件,在搜索过程中如果在当前位置(假设对应数字为i)填入除1以外的数字(假设为j),则可以在对应i+j位置同时填入j,这样最后得到的序列是满足第3个条件的。而为了满足字典序最大的条件,搜索数字时需要从n向2,从大到小搜索,这样得到的第一个序列就是题目所求序列。以上为基本思路。
在代码实现时,我踩过的坑主要有2个:
1.代码中递归调用到的地方有两处,其中第一处是当前位置部位不为空(即当前位置是之前搜索中已经填入的两个位置中的后面一个),这种情况可以直接搜索下一个位置,但是需要注意在当前递归函数搜索完成,进入下一步时需要先验证一下当前位置是否为空。如果不为空说明该位置仍然是之前已经填入过的位置,这种情况说明前面的尝试失败,应该直接返回,不进入之后的搜索过程。如果此处再进行修改则会导致不符合第三个条件。
2.在第二个递归调用的地方,如果当前选择的数字不是1,则应该同时复原两个位置为空。如果只复原一个同样会造成前后不一的情况
代码
class Solution:
def constructDistancedSequence(self, n: int) -> List[int]:
if n==1:
return [1]
vis=[False for i in range(2*n-1)]
temp=['' for i in range(2*n-1)]
ans=[]
flag=False
def dfs(cur):
nonlocal vis,n,ans,temp,flag
if flag:
return
if cur==2*n-1:
ans=copy.deepcopy(temp)
flag=True
return
if temp[cur]!='':
dfs(cur+1)
if temp[cur]!='':
return
for i in range(n,0,-1):
if not vis[i]:
if i==1:
temp[cur]=i
else:
if cur+i>=2*n-1:
continue
if temp[cur+i]!='':
continue
temp[cur]=i
temp[cur+i]=i
print(temp,i)
vis[i]=True
dfs(cur+1)
vis[i]=False
if i!=1:
temp[cur+i]=''
temp[cur]=''
vis[i]=0
dfs(0)
return ans