| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 896864 | Litusiano | 순열 (APIO22_perm) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include<bits/stdc++.h>
#include "perm.h"
using namespace std;
const int CT = 1000;
mt19937 rng((unsigned int) chrono::steady_clock::now().time_since_epoch().count());
vector<int> shuffle(vector<int>& v){
int sz = v.size();
for(int i = 0; i < 2 * v.size(); i++){
int a = rng()%sz; int b = rng()%sz;
swap(v[a],v[b]);
}
return v;
}
long long getdp(vector<int>& v, long long MX) {
vector<long long> dp(v.size() + 1, 0);
dp[0] = 1;
for (int x : v)
{
for (int i = 0; i <= x; i++)
{
dp[x+1] += dp[i];
dp[x+1] = min(dp[x+1],MX+1);
if(dp[x+1] == MX+1) return MX+1;
}
}
long long result = 0;
for (int i = 0; i <= (int)v.size(); i++){
result += dp[i];
result = min(result,MX+1);
}
return result;
}
vector<int> construct_permutation(long long k)
{
const int CT = 1000;
if(k <= 90){
vector<int> v;
for(int i = k-2; i>=0; i--){
v.push_back(i);
}
return v;
}
vector<int> v;
for(int a = 1; a<=90; a++){
v.push_back(a-1);
if(a >= 8){
for(int _ = 0; _ < CT; _++){
v = shuffle(v);
if(getdp(v,k) == k){
return v;
}
}
}
else{
sort(v.begin(),v.end());
do{
if(getdp(v,k) == k) return v;
}while(next_permutation(v.begin(),v.end()));
}
}
return v;
}
static long long MX=1e18;
static bool check_permutation(vector<int> v)
{
sort(v.begin(),v.end());
for(int i=0;i<v.size();i++)
if(v[i]!=i) return 0;
return 1;
}
long long count_increasing(const vector<int>& v) {
vector<long long> dp(v.size() + 1, 0);
dp[0] = 1;
for (int x : v)
{
for (int i = 0; i <= x; i++)
{
dp[x+1] += dp[i];
dp[x+1] = min(dp[x+1],MX+1);
}
}
long long result = 0;
for (int i = 0; i <= (int)v.size(); i++){
result += dp[i];
result = min(result,MX+1);
}
return result;
}
int main() {
int t;
assert(1 == scanf("%d", &t));
while(t--)
{
long long k;
assert(1 == scanf("%lld",&k));
vector<int> ret=construct_permutation(k);
// for(int i : ret) cout<<i<<" ";
// cout<<endl;
if(!check_permutation(ret))
{
printf("WA: Returned array is not a permutation\n");
exit(0);
}
long long inc=count_increasing(ret);
if(inc!=k)
{
if(inc==MX+1)
printf("WA: Expected %lld increasing subsequences, found more than %lld\n",k, MX);
else
printf("WA: Expected %lld increasing subsequences, found %lld\n",k,inc);
exit(0);
}
printf("%d\n",(int)ret.size());
for(int i=0;i<ret.size();i++)
{
printf("%d",ret[i]);
if(i+1==ret.size())
printf("\n");
else
printf(" ");
}
}
return 0;
}