結果

問題 No.1075 木の上の山
ユーザー xllend3xllend3
提出日時 2020-06-05 22:53:55
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 1,506 ms / 2,000 ms
コード長 3,438 bytes
コンパイル時間 1,915 ms
コンパイル使用メモリ 176,748 KB
実行使用メモリ 78,024 KB
最終ジャッジ日時 2024-12-17 17:11:04
合計ジャッジ時間 9,215 ms
ジャッジサーバーID
(参考情報)
judge5 / judge4
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 21 ms
54,600 KB
testcase_01 AC 21 ms
54,472 KB
testcase_02 AC 20 ms
54,596 KB
testcase_03 AC 20 ms
54,600 KB
testcase_04 AC 19 ms
54,600 KB
testcase_05 AC 20 ms
54,600 KB
testcase_06 AC 20 ms
54,472 KB
testcase_07 AC 22 ms
54,732 KB
testcase_08 AC 21 ms
54,856 KB
testcase_09 AC 21 ms
54,984 KB
testcase_10 AC 21 ms
54,980 KB
testcase_11 AC 20 ms
54,984 KB
testcase_12 AC 21 ms
54,860 KB
testcase_13 AC 22 ms
55,244 KB
testcase_14 AC 21 ms
55,112 KB
testcase_15 AC 22 ms
55,244 KB
testcase_16 AC 22 ms
55,112 KB
testcase_17 AC 55 ms
70,084 KB
testcase_18 AC 1,506 ms
77,636 KB
testcase_19 AC 63 ms
78,024 KB
testcase_20 AC 1,505 ms
77,772 KB
testcase_21 AC 65 ms
77,896 KB
testcase_22 AC 68 ms
77,768 KB
testcase_23 AC 70 ms
77,644 KB
testcase_24 AC 69 ms
77,768 KB
testcase_25 AC 70 ms
77,892 KB
testcase_26 AC 903 ms
77,764 KB
testcase_27 AC 426 ms
77,768 KB
testcase_28 AC 64 ms
77,892 KB
testcase_29 AC 387 ms
77,768 KB
testcase_30 AC 547 ms
77,772 KB
testcase_31 AC 384 ms
77,896 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include<bits/stdc++.h>
using namespace std;
#define X first
#define Y second
#define mp make_pair
#define pb push_back
#define rep(i,a,n) for(int i=(a);i<=(n);++i)
#define dep(i,a,n) for(int i=(a);i>=(n);--i)
#define eps 1e-8
#define pi 3.1415926535897
#define sqr(x) ((x)*(x))
#define MAX(a,b) a=max(a,b)
#define MIN(a,b) a=min(a,b)
#define SZ(x) ((int)(x).size())
#define CPY(a,b) memcpy(a,b,sizeof(a))
#define CLR(a) memset(a,0,sizeof(a))
#define POSIN(x,y) (1<=(x)&&(x)<=n&&1<=(y)&&(y)<=m)
#define all(x) (x).begin(),(x).end()
#define COUT(S,x) cout<<fixed<<setprecision(x)<<S<<endl
#define buli(x) (__builtin_popcountll(x))
#define bur0(x) (__builtin_ctzll(x))
#define bul2(x) (63-__builtin_clzll(x))
#define pw(x) ((ll(1))<<(x))
#define upmo(a,b) (((a)=((a)+(b))%mo)<0?(a)+=mo:(a))
#define mmo(a,b) (((a)=1ll*(a)*(b)%mo)<0?(a)+=mo:(a))
#define y0 y0z
#define y1 y1z
#define yn ynz
#define j0 j0z
#define j1 j1z
#define jn jnz
#define tm tmz
#ifdef LOCAL
#define debug(...) fprintf(stderr, __VA_ARGS__)
#else
#define debug(...) 
#endif
typedef long long ll;
typedef double lf;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;
typedef pair<lf,lf> pff;
typedef complex<double> CD;
const int inf=0x3f3f3f3f;
const int mo=1000000007;
inline void gn(long long&x){
	int sg=1;char c;while(((c=getchar())<'0'||c>'9')&&c!='-');c=='-'?(sg=-1,x=0):(x=c-'0');
	while((c=getchar())>='0'&&c<='9')x=x*10+c-'0';x*=sg;
}
inline void gn(int&x){long long t;gn(t);x=t;}
inline void gn(unsigned long long&x){long long t;gn(t);x=t;}
inline void gn(double&x){double t;scanf("%lf",&t);x=t;}
inline void gn(long double&x){double t;scanf("%lf",&t);x=t;}
template<class T1,class T2>inline void gn(T1&r,T2&s){gn(r),gn(s);}
template<class T1,class T2,class T3>inline void gn(T1&r,T2&s,T3&t){gn(r),gn(s),gn(t);}
template<class T1,class T2,class T3,class T4>inline void gn(T1&r,T2&s,T3&t,T4&u){gn(r),gn(s),gn(t),gn(u);}
inline void gs(char *s){scanf("%s",s);}
inline void gc(char &c){while((c=getchar())>126 || c<33);}
inline void pc(char c){putchar(c);}
const int DX[]={1,0,-1,0},DY[]={0,1,0,-1};
ll powmod(ll a,ll b) {ll res=1;a%=mo;for(;b;b>>=1){if(b&1)res=res*a%mo;a=a*a%mo;}return res;}
ll powmod(ll a,ll b,ll mo) {ll res=1;a%=mo;for(;b;b>>=1){if(b&1)res=res*a%mo;a=a*a%mo;}return res;}
ll gcd(ll a,ll b) { return b?gcd(b,a%b):a;}
//*******************************************

const int N=111111,M=111111;
int l,m,n,t,C,x,y;
vector<int>a[N];
int fa[N];
vector<ll>MA[1011][1011][2];
void dfs(int x,int f){
	for(auto i:a[x])if(i!=f)dfs(i,x);
	fa[x]=f;
}
void cal(int x,int y,int z){
	if(SZ(MA[x][y][z]))return;
	vector<ll>V(m+1);
	if(y==0){
		rep(i,0,m)V[i]=1;
	}else{
		for(auto i:a[y])if(i!=x)cal(y,i,z);
		rep(i,1,m)V[i]=1;
		for(auto i:a[y])if(i!=x){
			rep(j,1,m)V[j]=1ll*V[j]*MA[y][i][z][j]%mo;
		}
		rep(i,1,m)V[i]=(V[i]+V[i-1])%mo;
	}
	MA[x][y][z]=V;
}
ll calc(int x){
	cal(x,fa[x],0);
	cal(fa[x],x,1);
	ll ans=0;
	rep(i,1,m)ans=(ans+1ll*MA[x][fa[x]][0][i-1]*(MA[fa[x]][x][1][i]-MA[fa[x]][x][1][i-1]+mo))%mo;
	// printf("%d %lld\n",x,ans);
	return ans;
}
int main(){
#ifdef LOCAL
	freopen("F.in","r",stdin);//freopen("F.out","w",stdout);
#endif
	scanf("%d%d",&n,&m);
	rep(i,2,n)scanf("%d%d",&x,&y),a[x].pb(y),a[y].pb(x);
	dfs(1,0);
	ll ans=0;
	rep(i,1,n)ans=(ans+calc(i))%mo;
	ans=(ans+mo)%mo;
	cout<<ans<<endl;
	// for(auto i:MA){
	// 	printf("%d %d %d: ",i.X.X.X,i.X.X.Y,i.X.Y);
	// 	rep(j,0,m)printf("%lld ",i.Y[j]);puts("");
	// }
	return 0;
}
0