题意
称一个 的排列为波浪形的排列当且仅当
,
。
求 的第
个波浪形的排列。
解法1:DFS(TLE)
不妨把波列形的排列看成小虫的游走。小虫从 的任意点,任意方向出发,选择前方任意一个未访问过的点,到达后掉头。这样最终产生一个排列。
可以按字典序暴力搜索所有排列,第 个排列即为答案。
下图为 的DFS树:
代码
const int N=25;
class Solution {
public:
/**
*
* @param n int整型 木棒的个数
* @param k long长整型 第k个排列
* @return int整型vector
*/
int a[N];
bool vis[N];
bool flag;
long long cnt;
void dfs(int n, int k, int pos, bool dir) {
if (pos == n) { // 搜到一个排列
//for (int i = 0; i < n; ++i) printf("%d ", a[i]);
++cnt;
//cout << cnt << endl;
if (cnt == k) flag = true; // 搜到第 k 个排列
return;
}
if (pos == 0) {
for (int i = 1; i <= n; ++i) {
a[pos] = i;
vis[i] = true;
dfs(n, k, pos + 1, 0); // 先向左爬
vis[i] = false;
if (flag) return;
vis[i] = true;
dfs(n, k, pos + 1, 1); // 再向右爬
vis[i] = false;
if (flag) return;
}
} else if (!dir) {
int cnt = 0;
for (int i = 1; i < a[pos - 1]; ++i) {
if (vis[i]) continue;
a[pos] = i;
vis[i] = true;
dfs(n, k, pos + 1, 1); // 去左边的点向右爬
vis[i] = false;
if (flag) return;
}
} else {
int cnt = 0;
for (int i = a[pos - 1] + 1; i <= n; ++i) {
if (vis[i]) continue;
a[pos] = i;
vis[i] = true;
dfs(n, k, pos + 1, 0); // 去右边的点向左爬
vis[i] = false;
if (flag) return;
}
}
}
vector<int> stick(int n, long long k) {
// write code here
cnt = 0;
flag = false;
for (int i = 1; i <= n; ++i) vis[i] = false;
dfs(n, k, 0, 0);
vector<int> ret;
for (int i = 0; i < n; ++i) ret.push_back(a[i]);
return ret;
}
}; 复杂度分析
空间复杂度:保存dfs树链信息
时间复杂度:需要逐个枚举满足要求的排列。不太方便精确分析,但粗略估计个数在 到
之间,复杂度比
略低。
解法2:DP+DFS
记 为小虫前方有
个未访问的点,后方有
个未访问的点时的方案数。
有递推关系 。
而dfs树上每棵子树的size恰好与dp值相对应。利用size信息,我们便可以判断答案在那棵子树中。
dfs时,我们只需要搜索答案所在的那条链即可。
代码
const int N=25;
class Solution {
public:
/**
*
* @param n int整型 木棒的个数
* @param k long长整型 第k个排列
* @return int整型vector
*/
long long dp[N][N];
int a[N];
bool vis[N];
void init(){
int lim=20;
dp[0][0]=1;
for(int sum=1; sum<=lim; ++sum){
for(int i=0; i<=sum; ++i){
int j=sum-i;
dp[i][j]=0;
for(int k=0; k<i; ++k){
dp[i][j]+=dp[sum-1-k][k]; // 递推
}
}
}
for(int i=1; i<=lim; ++i) vis[i]=false;
}
void dfs(int n, int pos, long long cur, bool dir){
if(pos==n) return;
if(pos==0){
for(int i=1; i<=n; ++i){
long long sz=dp[i-1][n-i]; // 子树大小
if(cur>sz) cur-=sz; // 跳过子树
else{ // 进入子树
a[pos]=i;
vis[i]=true;
dfs(n, pos+1, cur, 0);
vis[i]=false;
break;
}
sz=dp[n-i][i-1];
if(cur>sz) cur-=sz;
else{
a[pos]=i;
vis[i]=true;
dfs(n, pos+1, cur, 1);
vis[i]=false;
return;
}
}
} else if(!dir){
int cnt=0;
for(int i=1; i<a[pos-1]; ++i){
if(vis[i]) continue;
cnt++;
long long sz=dp[n-pos-cnt][cnt-1];
if(cur>sz) cur-=sz;
else{
a[pos]=i;
vis[i]=true;
dfs(n, pos+1, cur, 1);
vis[i]=false;
return;
}
}
} else{
int cnt=0;
for(int i=1; i<=n; ++i){
if(vis[i]) continue;
cnt++;
if(i<=a[pos-1]) continue;
long long sz=dp[cnt-1][n-pos-cnt];
if(cur>sz) cur-=sz;
else{
a[pos]=i;
vis[i]=true;
dfs(n, pos+1, cur, 0);
vis[i]=false;
return;
}
}
}
}
vector<int> stick(int n, long long k) {
// write code here
init();
dfs(n, 0, k, 0);
vector<int> ret;
for(int i=0; i<n; ++i) ret.push_back(a[i]);
return ret;
}
}; 复杂度分析
空间复杂度:储存DP数组需要
时间复杂度:DP ,dfs只需在答案链上每个节点枚举
,共



京公网安备 11010502036488号