結果

問題 No.2846 Birthday Cake
ユーザー 沙耶花沙耶花
提出日時 2024-08-23 22:00:28
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.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
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 34
権限があれば一括ダウンロードができます

ソースコード

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