QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#456365#8529. Balance of Permutationarnold518AC ✓2114ms22608kbC++172.5kb2024-06-27 20:14:432024-06-27 20:14:44

Judging History

This is the latest submission verdict.

  • [2024-06-27 20:14:44]
  • Judged
  • Verdict: AC
  • Time: 2114ms
  • Memory: 22608kb
  • [2024-06-27 20:14:43]
  • Submitted

answer

#include <bits/stdc++.h>
#pragma GCC optimize("Ofast")
#pragma GCC target("avx,avx2,fma")
#define ff first
#define ss second
using namespace std;

typedef long long ll;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
typedef __int128 dll;

const int MAXN = 30;

int N, B, A[MAXN+10], AP[MAXN+10];
bool chk[MAXN+10], vis[MAXN+10];
dll K;

dll dp[MAXN+10][MAXN+10][MAXN*MAXN+10];
dll f[MAXN+10][MAXN+10], g[MAXN+10];

dll getdp(int h, int comp, int sum)
{
	if(!(1<=h && h<=N+1)) return 0;
	if(!(0<=comp && comp<=N)) return 0;
	if(!(0<=sum && sum<=N*N)) return 0;
	return dp[h][comp][sum];
}

dll solve(int p) // chosen for i<=p
{
	if(p==N+1) return 1;
	int val=0;
	int cnt=0, cnt2=0;
	for(int i=1; i<=N; i++) vis[i]=0;
	for(int i=1; i<=p; i++)
	{
		if(A[i]>p)
		{
			cnt++;
			val+=abs(p+1-i);
			vis[A[i]]=1;
		}
		else val+=abs(A[i]-i);

		if(AP[i]==0)
		{
			val+=abs(p+1-i);
		}
	}
	cnt2=cnt;
	// printf("!%d %d\n", cnt, val);

	dp[N+1][0][B-val]=1;
	cnt=0;
	for(int h=N; h>p; h--)
	{
		if(vis[h]) cnt++;
		for(int comp=cnt; comp<=N; comp++)
		{
			for(int sum=0; sum<=N*N; sum++)
			{
				dp[h][comp][sum]=0;

				if(!vis[h])
				{
					dp[h][comp][sum]+=getdp(h+1, comp, sum+2*comp);
					dp[h][comp][sum]+=getdp(h+1, comp+1, sum+2*(comp+1))*(comp+1);
					dp[h][comp][sum]+=getdp(h+1, comp, sum+2*comp)*(comp*2-cnt);
					dp[h][comp][sum]+=getdp(h+1, comp-1, sum+2*(comp-1))*(comp-cnt);
				}
				else
				{
					dp[h][comp][sum]+=getdp(h+1, comp, sum+2*comp);
					dp[h][comp][sum]+=getdp(h+1, comp-1, sum+2*(comp-1));
				}
				// printf("%d %d %d : %lld\n", h, comp, sum, (ll)dp[h][comp][sum]);
			}
		}
	}
	dp[N+1][0][B-val]=0;

	dll ret=dp[p+1][cnt2][0]*g[cnt2];
	// for(int i=1; i<=p; i++) printf("%d ", A[i]); printf("\n");
	// for(int i=1; i<=p; i++) printf("%d ", AP[i]); printf("\n");
	// printf(" : %lld\n\n", (ll)ret);
	return ret;

}

int main() {
	cin.tie(0); ios_base::sync_with_stdio(0);
	
	string SK;
	cin>>N>>B>>SK;
	for(int i=0; i<SK.size(); i++) K=K*10+(SK[i]-'0');

	f[1][1]=1;
	for(int i=2; i<=N; i++)
	{
		for(int j=1; j<=i; j++)
		{
			f[i][j]=f[i-1][j-1]+f[i-1][j]*(i-1);
		}
	}
	g[0]=1;
	for(int i=1; i<=N; i++) for(int j=1; j<=i; j++) g[i]+=f[i][j];

	for(int i=1; i<=N; i++)
	{
		for(int j=1; j<=N; j++)
		{
			if(chk[j]) continue;
			A[i]=j;
			AP[j]=i;
			dll t=solve(i);
			if(t<K)
			{
				K-=t;
				AP[j]=0;
			}
			else
			{
				chk[j]=true;
				break;
			}
		}
	}
	for(int i=1; i<=N; i++) printf("%d ", A[i]); printf("\n");
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 0ms
memory: 7964kb

input:

6 6 6

output:

1 2 6 3 4 5 

result:

ok 6 numbers

Test #2:

score: 0
Accepted
time: 1076ms
memory: 22304kb

input:

30 300 3030303030303030303030

output:

1 2 3 4 9 23 20 28 24 16 21 17 27 29 8 26 25 30 19 18 22 12 7 13 6 10 5 15 14 11 

result:

ok 30 numbers

Test #3:

score: 0
Accepted
time: 0ms
memory: 6072kb

input:

1 0 1

output:

1 

result:

ok 1 number(s): "1"

Test #4:

score: 0
Accepted
time: 1ms
memory: 5824kb

input:

2 0 1

output:

1 2 

result:

ok 2 number(s): "1 2"

Test #5:

score: 0
Accepted
time: 1ms
memory: 5820kb

input:

2 2 1

output:

2 1 

result:

ok 2 number(s): "2 1"

Test #6:

score: 0
Accepted
time: 1ms
memory: 5876kb

input:

5 8 3

output:

1 5 4 2 3 

result:

ok 5 number(s): "1 5 4 2 3"

Test #7:

score: 0
Accepted
time: 0ms
memory: 7968kb

input:

7 20 100

output:

3 6 7 4 1 5 2 

result:

ok 7 numbers

Test #8:

score: 0
Accepted
time: 1ms
memory: 8160kb

input:

7 2 6

output:

2 1 3 4 5 6 7 

result:

ok 7 numbers

Test #9:

score: 0
Accepted
time: 1ms
memory: 8240kb

input:

7 24 1

output:

4 5 6 7 1 2 3 

result:

ok 7 numbers

Test #10:

score: 0
Accepted
time: 1ms
memory: 7968kb

input:

7 22 360

output:

7 6 4 3 5 2 1 

result:

ok 7 numbers

Test #11:

score: 0
Accepted
time: 1ms
memory: 8008kb

input:

7 20 358

output:

5 7 2 4 6 3 1 

result:

ok 7 numbers

Test #12:

score: 0
Accepted
time: 3ms
memory: 10328kb

input:

10 48 10001

output:

7 5 8 9 6 10 3 4 1 2 

result:

ok 10 numbers

Test #13:

score: 0
Accepted
time: 0ms
memory: 9980kb

input:

10 42 10101

output:

3 9 6 8 10 5 7 2 1 4 

result:

ok 10 numbers

Test #14:

score: 0
Accepted
time: 458ms
memory: 18488kb

input:

25 300 1

output:

7 14 15 16 17 18 19 20 21 22 23 24 25 1 2 3 4 5 6 8 9 10 11 12 13 

result:

ok 25 numbers

Test #15:

score: 0
Accepted
time: 707ms
memory: 18180kb

input:

25 300 283788388040048639877

output:

25 24 23 22 21 20 19 18 17 16 11 12 13 14 15 10 9 8 7 5 6 4 2 1 3 

result:

ok 25 numbers

Test #16:

score: 0
Accepted
time: 757ms
memory: 18704kb

input:

26 302 105773752969551707419545

output:

19 22 25 13 17 18 23 20 10 26 16 6 5 11 14 12 24 4 3 21 1 15 7 8 2 9 

result:

ok 26 numbers

Test #17:

score: 0
Accepted
time: 846ms
memory: 20460kb

input:

27 308 8781128321749037280676555

output:

16 18 17 21 25 6 20 24 22 15 27 5 7 8 2 9 26 13 1 3 14 10 23 19 4 11 12 

result:

ok 27 numbers

Test #18:

score: 0
Accepted
time: 974ms
memory: 20228kb

input:

28 304 806517199954337651602356955

output:

12 17 5 16 23 26 25 15 20 2 19 7 22 24 6 13 11 10 28 8 1 21 18 14 27 3 4 9 

result:

ok 28 numbers

Test #19:

score: 0
Accepted
time: 1270ms
memory: 20184kb

input:

29 322 40281026669581503094652149519

output:

16 21 10 25 17 29 9 28 2 8 26 27 22 4 3 5 18 14 19 1 23 20 15 11 13 7 6 12 24 

result:

ok 29 numbers

Test #20:

score: 0
Accepted
time: 1775ms
memory: 22320kb

input:

30 400 46479902466857426153849991132

output:

25 19 30 29 9 20 26 21 14 27 28 10 22 11 24 2 7 4 18 17 5 13 12 6 8 1 15 23 16 3 

result:

ok 30 numbers

Test #21:

score: 0
Accepted
time: 1527ms
memory: 22276kb

input:

30 450 1140008168482799670544355

output:

26 16 17 18 19 20 21 22 23 24 25 27 28 29 30 1 2 3 5 9 4 8 14 10 6 11 12 15 7 13 

result:

ok 30 numbers

Test #22:

score: 0
Accepted
time: 804ms
memory: 22228kb

input:

30 150 480087379811286955791425915

output:

7 4 8 5 16 3 1 12 13 11 9 10 15 25 18 17 20 30 28 2 6 14 23 21 24 26 27 22 19 29 

result:

ok 30 numbers

Test #23:

score: 0
Accepted
time: 802ms
memory: 22508kb

input:

30 150 480087379811286955791439470

output:

7 4 8 5 16 3 1 12 13 11 9 10 15 25 18 17 20 30 28 2 19 6 22 24 21 23 26 14 29 27 

result:

ok 30 numbers

Test #24:

score: 0
Accepted
time: 1713ms
memory: 22516kb

input:

30 440 41509275104334759322587324

output:

22 23 20 24 18 30 19 26 21 28 4 29 17 25 27 16 3 1 2 5 8 13 10 15 7 12 9 14 11 6 

result:

ok 30 numbers

Test #25:

score: 0
Accepted
time: 1533ms
memory: 22336kb

input:

30 450 1140008168482800727111311

output:

26 16 17 18 19 20 21 22 23 24 25 27 28 29 30 1 2 5 7 14 4 15 8 11 3 13 10 9 6 12 

result:

ok 30 numbers

Test #26:

score: 0
Accepted
time: 1808ms
memory: 22280kb

input:

30 400 52289890275214604423031772929

output:

26 27 29 21 28 16 18 11 2 25 24 23 6 30 20 13 17 10 15 4 9 12 8 22 19 1 5 7 3 14 

result:

ok 30 numbers

Test #27:

score: 0
Accepted
time: 170ms
memory: 22480kb

input:

30 0 1

output:

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 

result:

ok 30 numbers

Test #28:

score: 0
Accepted
time: 1416ms
memory: 22316kb

input:

30 450 1

output:

16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 

result:

ok 30 numbers

Test #29:

score: 0
Accepted
time: 2114ms
memory: 22232kb

input:

30 450 1710012252724199424000000

output:

30 29 28 27 26 25 24 23 22 21 20 19 18 17 16 15 14 13 12 11 10 9 8 7 6 5 4 3 2 1 

result:

ok 30 numbers

Test #30:

score: 0
Accepted
time: 1942ms
memory: 20904kb

input:

30 450 1692383260428073656742269

output:

30 27 26 28 18 29 21 19 25 17 20 16 24 22 23 7 13 4 6 3 5 12 1 15 14 9 11 8 2 10 

result:

ok 30 numbers

Test #31:

score: 0
Accepted
time: 1910ms
memory: 22508kb

input:

30 302 5918364042599361729860937331200

output:

30 29 28 27 26 25 14 8 9 10 11 12 13 7 15 16 17 18 19 20 21 22 23 24 6 5 4 3 2 1 

result:

ok 30 numbers

Test #32:

score: 0
Accepted
time: 1212ms
memory: 22608kb

input:

30 254 2256781660157136563723839089600

output:

25 2 3 12 7 16 19 8 22 6 11 17 27 26 10 24 15 21 20 18 28 9 30 23 14 13 5 29 4 1 

result:

ok 30 numbers

Test #33:

score: 0
Accepted
time: 1834ms
memory: 20776kb

input:

30 448 3131906441000512625049600

output:

23 20 28 18 26 30 19 29 27 22 17 24 21 25 2 13 16 15 14 12 11 10 9 8 7 6 5 4 3 1 

result:

ok 30 numbers

Test #34:

score: 0
Accepted
time: 186ms
memory: 20328kb

input:

30 2 20

output:

1 2 3 4 5 6 7 8 9 11 10 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 

result:

ok 30 numbers

Test #35:

score: 0
Accepted
time: 185ms
memory: 22252kb

input:

30 2 29

output:

2 1 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 

result:

ok 30 numbers

Extra Test:

score: 0
Extra Test Passed