結果

問題 No.2846 Birthday Cake
ユーザー 沙耶花沙耶花
提出日時 2024-08-23 22:00:28
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 1,526 ms / 2,000 ms
コード長 1,423 bytes
コンパイル時間 4,763 ms
コンパイル使用メモリ 274,512 KB
実行使用メモリ 72,832 KB
最終ジャッジ日時 2024-08-23 22:00:55
合計ジャッジ時間 21,861 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,812 KB
testcase_01 AC 1 ms
6,940 KB
testcase_02 AC 932 ms
57,216 KB
testcase_03 AC 2 ms
6,944 KB
testcase_04 AC 2 ms
6,940 KB
testcase_05 AC 1,061 ms
63,872 KB
testcase_06 AC 1,171 ms
68,352 KB
testcase_07 AC 1,271 ms
71,040 KB
testcase_08 AC 1,266 ms
72,192 KB
testcase_09 AC 1,281 ms
72,704 KB
testcase_10 AC 1,385 ms
72,832 KB
testcase_11 AC 1,435 ms
72,832 KB
testcase_12 AC 660 ms
55,936 KB
testcase_13 AC 871 ms
69,120 KB
testcase_14 AC 2 ms
6,940 KB
testcase_15 AC 12 ms
6,944 KB
testcase_16 AC 44 ms
9,472 KB
testcase_17 AC 1,022 ms
70,400 KB
testcase_18 AC 1,526 ms
72,832 KB
testcase_19 AC 2 ms
6,940 KB
testcase_20 AC 2 ms
6,940 KB
testcase_21 AC 2 ms
6,940 KB
testcase_22 AC 8 ms
6,944 KB
testcase_23 AC 7 ms
6,940 KB
testcase_24 AC 938 ms
70,272 KB
testcase_25 AC 2 ms
6,940 KB
testcase_26 AC 4 ms
6,940 KB
testcase_27 AC 6 ms
6,940 KB
testcase_28 AC 2 ms
6,944 KB
testcase_29 AC 12 ms
6,940 KB
testcase_30 AC 2 ms
6,944 KB
testcase_31 AC 3 ms
6,940 KB
testcase_32 AC 161 ms
19,840 KB
testcase_33 AC 901 ms
70,272 KB
testcase_34 AC 39 ms
9,472 KB
testcase_35 AC 41 ms
9,472 KB
testcase_36 AC 295 ms
25,856 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <stdio.h>
#include <bits/stdc++.h>
#include <atcoder/all>
using namespace atcoder;
using mint = modint998244353;
using namespace std;
#define rep(i,n) for (int i = 0; i < (n); ++i)
#define Inf32 1000000001
#define Inf64 1000000000000000001

long long N,K;

long long En = 1;
vector<long long> la,lb;
long long L;
map<long long,__int128> mp[25];
map<long long,__int128> mp2[25];
void dfs(int ci,int ck,long long cx,__int128 cv){
	if(cx > L)return;

	if(ck>=la.size()){
		__int128 t = 1;
		rep(i,K){
			t *= i+1;
		}
		mp[ci][cx] += t/cv;
		return;
	}
	__int128 t = 1LL;
	rep(i,K-ci+1){
		if(i==0){
			dfs(ci,ck+1,cx,cv);
		}
		else{
			long long ncx = cx;
			ncx += (L / la[ck])*i;
			dfs(ci+i,ck+1,ncx,cv*t);
		}
		t *= i+1;
	}
}

int main(){
	
	cin>>K>>N;
	rep(i,K)En *= i+1;
	L = 1;
	rep(i,N)L = lcm(L,i+1);
	rep(i,N){
		if(i%2)la.push_back(i+1);
		else lb.push_back(i+1);
	}
	dfs(0,0,0,1);
	swap(la,lb);
	swap(mp,mp2);
	dfs(0,0,0,1);
	__int128 ans  =0;
	rep(i,K+1){
		for(auto x:mp[i]){
			if(!mp2[K-i].count(L-x.first))continue;
			__int128 A = x.first,B = x.second;
			__int128 X = B;
			__int128 Y = mp2[K-i][L-A];
			__int128 rem = 1LL;
			rep(j,K+1){
				if(j==0)continue;
				int cc = j;
				if(X%j==0){
					X /= j;
					continue;
				}
				if(Y%j==0){
					Y /= j;
					continue;
				}
				rem *= j;
			}
			X *= Y;
			X /= rem;
			ans += X;
		}
	}
	long long aa = ans;
	cout<<aa<<endl;
    return 0;
}
0