結果
問題 | No.196 典型DP (1) |
ユーザー | kmjp |
提出日時 | 2015-04-26 22:24:33 |
言語 | C++11 (gcc 13.3.0) |
結果 |
AC
|
実行時間 | 151 ms / 2,000 ms |
コード長 | 1,466 bytes |
コンパイル時間 | 1,277 ms |
コンパイル使用メモリ | 163,056 KB |
実行使用メモリ | 98,448 KB |
最終ジャッジ日時 | 2024-07-05 02:58:30 |
合計ジャッジ時間 | 4,234 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 41 |
ソースコード
#include <bits/stdc++.h> using namespace std; typedef signed long long ll; #undef _P #define _P(...) (void)printf(__VA_ARGS__) #define FOR(x,to) for(x=0;x<to;x++) #define FORR(x,arr) for(auto& x:arr) #define ITR(x,c) for(__typeof(c.begin()) x=c.begin();x!=c.end();x++) #define ALL(a) (a.begin()),(a.end()) #define ZERO(a) memset(a,0,sizeof(a)) #define MINUS(a) memset(a,0xff,sizeof(a)) //------------------------------------------------------- int N,K; vector<int> E[2020]; ll val[2020][2020]; ll mo=1000000007; int num[2020]; void dfs(int cur,int pre) { num[cur]=1; ll nn[2][2020]; int i,j,x; ZERO(nn); FOR(i,E[cur].size()) if(E[cur][i]==pre) break; if(i<E[cur].size()) E[cur].erase(E[cur].begin()+i); nn[0][0]=1; FOR(x,E[cur].size()) { int cu=x%2,ta=cu^1; int tar=E[cur][x]; ZERO(nn[ta]); dfs(tar,cur); for(i=0;i<=num[tar];i++) if(val[tar][i]) { for(j=0;j<=num[cur];j++) nn[ta][i+j] += nn[cu][j]*val[tar][i]%mo; } num[cur]+=num[tar]; FOR(i,num[cur]+1) nn[ta][i] %= mo; } FOR(i,2010) val[cur][i]=nn[E[cur].size()%2][i]; val[cur][num[cur]]=1; } void solve() { int i,j,k,l,r,x,y; string s; cin>>N>>K; FOR(i,N-1) { cin>>x>>y; E[x].push_back(y); E[y].push_back(x); } dfs(0,-1); cout<<val[0][K]<<endl; } int main(int argc,char** argv){ string s;int i; if(argc==1) ios::sync_with_stdio(false); FOR(i,argc-1) s+=argv[i+1],s+='\n'; FOR(i,s.size()) ungetc(s[s.size()-1-i],stdin); solve(); return 0; }