結果

問題 No.2846 Birthday Cake
コンテスト
ユーザー kotatsugame
提出日時 2024-08-23 21:29:01
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,171 ms / 2,000 ms
コード長 752 bytes
コンパイル時間 770 ms
コンパイル使用メモリ 79,604 KB
実行使用メモリ 59,776 KB
最終ジャッジ日時 2024-08-23 21:29:09
合計ジャッジ時間 7,575 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 34
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<vector>
#include<map>
#include<cassert>
using namespace std;
const long L=5354228880;
long l[25];
int K,N;
long fac[25];
long comb[25][25];
map<long,long>memo[25][25];
long dfs(int n,int k,long sum)
{
	if(l[n]*k<sum)return 0;
	if(n==N)return l[n]*k==sum;
	if(memo[n][k].find(sum)!=memo[n][k].end())return memo[n][k][sum];
	long ret=0;
	for(int i=0;i<=k&&l[n]*i+l[N]*(k-i)<=sum;i++)
	{
		ret+=comb[k][i]*dfs(n+1,k-i,sum-l[n]*i);
	}
	return memo[n][k][sum]=ret;
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	cin>>K>>N;
	for(int i=0;i<=K;i++)
	{
		comb[i][0]=comb[i][i]=1;
		for(int j=1;j<i;j++)comb[i][j]=comb[i-1][j-1]+comb[i-1][j];
	}
	for(int n=1;n<=N;n++)l[n]=L/n;
	long v=dfs(1,K,L);
	cout<<v<<endl;
}
0