P8376 [APIO2022] 排列

最新推荐文章于 2025-12-15 09:15:21 发布

原创最新推荐文章于 2025-12-15 09:15:21 发布 · 449 阅读

8 ·

CC 4.0 BY-SA版权

文章标签：

#c++ #算法 #开发语言

文章描述了一个名为`construct_permutation`的C++函数，用于生成给定整数k的排列，使用位操作和排序算法实现。函数通过构建并调整数组a来构造排列，最后对结果进行排序和调整索引。

#include <vector>
std::vector<int> construct_permutation(long long k);
//We'll be counting stars.
#include<bits/stdc++.h>
using namespace std;
#define pb emplace_back
#define For(i,j,k) for(int i=(j),i##_=(k);i<=i##_;i++)
#define Rof(i,j,k) for(int i=(j),i##_=(k);i>=i##_;i--)
#define ckmx(a,b) a=max(a,b)
#define ckmn(a,b) a=min(a,b)
#define debug cerr<<"Line#"<<__LINE__<<endl
#define vi vector<int>
const int inf=2022;
vi s,res;
int a[100];
bool b[100];
vi construct_permutation(long long k){
	res.clear(); 
	int x=1; while((1ll<<(x+1))<=k) x++;
	k-=(1ll<<x);
	memset(a,-1,sizeof a);
	For(i,0,x-1) b[i]=1ll&(k>>i);
	int mn=-1,se=-1;
	Rof(i,x-1,0) if(b[i]){
		if(i && b[i-1] && mn!=-1 && se!=-1){
			a[mn]--;
			a[i-1]=a[se];
			a[se]--;
			i--;
		}else if(mn!=-1){
			a[i]=a[mn]-1;
			se=mn;
			mn=i;
		}else{
			a[i]=inf-1;
			mn=i;
		}
	}
	For(i,0,x-1){
		if(a[i]!=-1) res.pb(a[i]);
		res.pb(2*inf-i);
	}
	reverse(res.begin(),res.end());
	s=res;
	sort(s.begin(),s.end());
	for(int &i:res) i=lower_bound(s.begin(),s.end(),i)-s.begin();
	return res;
}