hdu6628 permutation 1 (dfs)

题目链接
大意:给你两个数n,k,让你求出n的所有全排列中差异序列字典序第k小的排列
差异序列为 p i = a i + 1 a i , i [ 1 , n ) p_i=a_{i+1}-a_i,i\in[1,n) pi=ai+1ai,i[1,n)
思路:直接dfs枚举差异序列从小往大搜,然后对不合法的情况去掉(差异序列中元素的最大-最小<n-1才合法,重复的也不合法)
细节见代码:

#include<bits/stdc++.h>

#define LL long long
#define fi first
#define se second
#define mp make_pair
#define pb push_back

using namespace std;

LL gcd(LL a,LL b){return b?gcd(b,a%b):a;}
LL lcm(LL a,LL b){return a/gcd(a,b)*b;}
LL powmod(LL a,LL b,LL MOD){LL ans=1;while(b){if(b%2)ans=ans*a%MOD;a=a*a%MOD;b/=2;}return ans;}
const int N = 503;
int t,n,k;
int vis[N],p[N];
bool dfs(int now,int pre,int l,int r){
	if(now==n){
		if(k==1){
			for(int i=0;i<n;i++){
				cout<<p[i]-l+1;
				if(i<n-1)cout<<' ';
				else cout<<'\n';
			}
			return 1;
		}
		k--;
		return 0;
	}
	for(int i=1-n;i<=n-1;i++){//枚举差异序列字典序最小
		if(!vis[i+pre]){
			vis[i+pre]=1;
			if(max(i+pre,r)-min(l,i+pre)<=n-1){
				p[now]=i+pre;
				if(dfs(now+1,i+pre,min(i+pre,l),max(i+pre,r))){
					vis[i+pre]=0;
					return 1;
				}
			}
			vis[i+pre]=0;
		}
	}
	return 0;
}
int main(){
	ios::sync_with_stdio(false);
	for(cin>>t;t;t--){
		cin>>n>>k;int sta=0;
		vis[n]=1;
		p[0]=n;
		dfs(1,n,n,n);//相对大小
		vis[n]=0;
	}	
	return 0;
}


全部评论

相关推荐

10-07 23:57
已编辑
电子科技大学 Java
八街九陌:博士?客户端?开发?啊?
点赞 评论 收藏
分享
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务