結果
問題 | No.1075 木の上の山 |
ユーザー |
![]() |
提出日時 | 2020-06-05 22:33:50 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 154 ms / 2,000 ms |
コード長 | 2,300 bytes |
コンパイル時間 | 2,207 ms |
コンパイル使用メモリ | 129,464 KB |
最終ジャッジ日時 | 2025-01-10 22:44:54 |
ジャッジサーバーID (参考情報) |
judge3 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 30 |
ソースコード
#include <cstdio> #include <cstring> #include <iostream> #include <string> #include <cmath> #include <bitset> #include <vector> #include <map> #include <set> #include <queue> #include <deque> #include <algorithm> #include <complex> #include <unordered_map> #include <unordered_set> #include <random> #include <cassert> #include <fstream> #include <utility> #include <functional> #include <time.h> #include <stack> #include <array> #include <list> #define popcount __builtin_popcount using namespace std; typedef long long int ll; typedef pair<int, int> P; const ll MOD=1e9+7; int n, k; ll dp[1010][1010]; ll s[1010][1010]; vector<int> g[1010]; void dfs(int x, int p){ for(auto y:g[x]){ if(y==p) continue; dfs(y, x); } for(int i=1; i<=k; i++){ dp[x][i]=1; for(auto y:g[x]){ if(y==p) continue; (dp[x][i]*=s[y][i])%=MOD; } } for(int i=1; i<=k; i++){ s[x][i]=(s[x][i-1]+dp[x][i])%MOD; } } ll dp2[1010][1010]; void dfs2(int x, int p){ for(int i=1; i<=k; i++){ vector<ll> pl(1+g[x].size()), pr(1+g[x].size()); pl[0]=1; for(int j=0; j<g[x].size(); j++){ int y=g[x][j]; if(y==p){ pl[j+1]=pl[j]*dp2[x][i]%MOD; }else{ pl[j+1]=pl[j]*s[y][i]%MOD; } } pr[g[x].size()]=1; for(int j=(int)g[x].size()-1; j>=0; j--){ int y=g[x][j]; if(y==p){ pr[j]=pr[j+1]*dp2[x][i]%MOD; }else{ pr[j]=pr[j+1]*s[y][i]%MOD; } } for(int j=0; j<g[x].size(); j++){ int y=g[x][j]; if(y==p) continue; dp2[y][i]=(dp2[y][i-1]+pl[j]*pr[j+1])%MOD; } } for(auto y:g[x]){ if(y==p) continue; dfs2(y, x); } } int main() { cin>>n>>k; for(int i=0; i<n-1; i++){ int a, b; cin>>a>>b;a--; b--; g[a].push_back(b); g[b].push_back(a); } dfs(0, -1); for(int i=1; i<=k; i++) dp2[0][i]=1; dfs2(0, -1); ll ans=0; for(int i=0; i<n; i++){ for(int j=1; j<=k; j++){ if(i==0) (ans+=dp[i][j])%=MOD; else (ans+=dp[i][j]*dp2[i][j-1])%=MOD; } } cout<<ans<<endl; return 0; }