結果

問題 No.196 典型DP (1)
ユーザー piyoko_212
提出日時 2015-12-28 18:58:09
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 71 ms / 2,000 ms
コード長 894 bytes
コンパイル時間 357 ms
コンパイル使用メモリ 42,576 KB
実行使用メモリ 34,176 KB
最終ジャッジ日時 2024-09-19 07:56:30
合計ジャッジ時間 2,984 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 41
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:36:22: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   36 |         int a,b;scanf("%d%d",&a,&b);
      |                 ~~~~~^~~~~~~~~~~~~~
main.cpp:39:22: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   39 |                 scanf("%d%d",&p,&q);
      |                 ~~~~~^~~~~~~~~~~~~~

ソースコード

diff # 

#include<stdio.h>
#include<algorithm>
#include<vector>
using namespace std;
vector<int>g[2100];
int sz[2100];
int mod=1000000007;
long long dp[2100][2100];
long long dp2[2100][2100];
void dfs(int a,int b){
	sz[a]=1;
	for(int i=0;i<g[a].size();i++){
		if(b!=g[a][i]){
			dfs(g[a][i],a);
			sz[a]+=sz[g[a][i]];
		}
	}
	int now=0;
	dp2[0][0]=1;
	int cnt=0;
	for(int i=0;i<g[a].size();i++){
		if(b==g[a][i])continue;
		for(int j=0;j<=now+sz[g[a][i]];j++)dp2[cnt+1][j]=0;
		for(int j=0;j<=now;j++){
			for(int k=0;k<=sz[g[a][i]];k++){
				dp2[cnt+1][j+k]=(dp2[cnt+1][j+k]+dp2[cnt][j]*dp[g[a][i]][k])%mod;
			}
		}
		cnt++;
		now+=sz[g[a][i]];
	}
	for(int i=0;i<sz[a];i++)dp[a][i]=dp2[cnt][i];
	dp[a][sz[a]]++;
}
int main(){
	int a,b;scanf("%d%d",&a,&b);
	for(int i=0;i<a-1;i++){
		int p,q;
		scanf("%d%d",&p,&q);
		g[p].push_back(q);
		g[q].push_back(p);
	}
	dfs(0,-1);
	printf("%lld\n",dp[0][b]);
}
0