結果

問題 No.1227 I hate ThREE
ユーザー leaf_1415
提出日時 2020-09-11 22:09:31
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 150 ms / 2,000 ms
コード長 1,648 bytes
コンパイル時間 983 ms
コンパイル使用メモリ 86,504 KB
実行使用メモリ 62,300 KB
最終ジャッジ日時 2025-01-02 05:33:12
合計ジャッジ時間 3,879 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <cstdio>
#include <cmath>
#include <ctime>
#include <cstdlib>
#include <cassert>
#include <vector>
#include <list>
#include <stack>
#include <queue>
#include <deque>
#include <map>
#include <set>
#include <bitset>
#include <string>
#include <algorithm>
#include <utility>
#define llint long long
#define PI 3.1415926535897932384626433832795028841971
#define inf 1e18
#define rep(x, s, t) for(llint (x) = (s); (x) < (t); (x)++)
#define Rep(x, s, t) for(llint (x) = (s); (x) <= (t); (x)++)
#define chmin(x, y) (x) = min((x), (y))
#define chmax(x, y) (x) = max((x), (y))
#define mod 1000000007

using namespace std;
typedef pair<llint, llint> P;

llint n, C, M;
vector<llint> G[1005];
llint dp[1005][7505];

void dfs(int v, int p)
{
	for(int i = 0; i < G[v].size(); i++){
		if(G[v][i] == p) continue;
		dfs(G[v][i], v);
	}
	
	for(int j = 1; j <= M; j++){
		dp[v][j] = 1;
		for(int i = 0; i < G[v].size(); i++){
			llint u = G[v][i];
			if(u == p) continue;
			llint sum = 0;
			if(j+3 <= M) (sum += dp[u][j+3]) %= mod;
			if(j-3 >= 1) (sum += dp[u][j-3]) %= mod;
			dp[v][j] *= sum, dp[v][j] %= mod;
		}
	}
}

int main(void)
{
	ios::sync_with_stdio(0);
	cin.tie(0);
	
	cin >> n >> C;
	M = min(C, 7000LL);
	llint u, v;
	for(int i = 1; i <= n-1; i++){
		cin >> u >> v;
		G[u].push_back(v);
		G[v].push_back(u);
	}
	dfs(1, -1);
	
	/*for(int i = 1; i <= n; i++){
		for(int j = 1; j <= M; j++){
			cout << dp[i][j] << " ";
		}
		cout << endl;
	}*/
	
	llint ans = 0;
	for(int i = 1; i <= M; i++) ans += dp[1][i], ans %= mod;
	if(C > 7000) ans += dp[1][3500] * (C - 7000) % mod, ans %= mod;
	
	cout << ans << endl;
	
	return 0;
}
0