Submission #1676953


Source Code Expand

#include<iostream>
#include<vector>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<cstdlib>
#include<ctime>
#include<queue>
#include<set>
#include<map>
#include<stack>
#include<bitset>
using namespace std;
typedef long long LL;
const int N=2010;
int gi() {
	int w=0;bool q=1;char c=getchar();
	while ((c<'0'||c>'9') && c!='-') c=getchar();
	if (c=='-') q=0,c=getchar();
	while (c>='0'&&c <= '9') w=w*10+c-'0',c=getchar();
	return q? w:-w;
}
vector<int>e[N];
int f[N][N],fa[N],g[N];
int D,T;
const int mod=1e9+7;
inline void dfs(int k,int d) {
	if (d>D) D=d,T=k;
	for (int t:e[k]) if (t!=fa[k]) fa[t]=k,dfs(t,d+1);
}
inline void dp(int k,int fa,int d) {
	if (e[k].size()==1)
		for (int i=0;i<=D;i++)
			f[k][D+i]=f[k][D-i]=i<=d;
	else {
		for (int t:e[k]) if (t!=fa) dp(t,k,d-1);
		for (int i=0;i<=D*2;i++) {
			f[k][i]=1;
			for (int t:e[k])
				if (t!=fa)
					f[k][i]=1LL*f[k][i]*((i?f[t][i-1]:0)+(i==D*2?0:f[t][i+1]))%mod;
		}
	}
}
int main()
{
	int i,k,a,b,n=gi(),ans=0;
	for (i=1;i<n;i++) {
		a=gi(),b=gi();
		e[a].push_back(b);
		e[b].push_back(a);
	}
	D=0;dfs(1,0);
	fa[T]=0;D=0;dfs(a=T,0);
	if (D&1) {
		for (a=T,i=(++D)/=2;i--;a=fa[b=a]);
		dp(a,b,D);
		dp(b,a,D-1);
		for (i=0;i<=D*2;i++) ans=(ans+1LL*f[a][i]*f[b][i+1]+1LL*f[a][i+1]*f[b][i])%mod;
		dp(a,b,D-1);
		dp(b,a,D);
		for (i=0;i<=D*2;i++) ans=(ans+1LL*f[a][i]*f[b][i+1]+1LL*f[a][i+1]*f[b][i])%mod;
		dp(a,b,D-1);
		dp(b,a,D-1);
		for (i=0;i<=D*2;i++) {
			ans=(ans-2LL*f[a][i]*f[b][i])%mod;
			ans=(ans-1LL*f[a][i]*f[b][i+2])%mod;
			ans=(ans-1LL*f[a][i+2]*f[b][i])%mod;
		}
		if (ans<0) ans+=mod;
	} else {
		for (k=T,i=D/=2;i--;k=fa[k]);
		dp(k,0,D);
		for (i=0;i<=D*2;i++) (ans+=f[k][i])%=mod;
	}
	printf("%d\n",ans);
	return 0;
}

Submission Info

Submission Time
Task D - Oriented Tree
User laofu
Language C++14 (GCC 5.4.1)
Score 1800
Code Size 1816 Byte
Status AC
Exec Time 18 ms
Memory 8576 KB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 1800 / 1800
Status
AC × 4
AC × 33
Set Name Test Cases
Sample 0_00.txt, 0_01.txt, 0_02.txt, 0_03.txt
All 0_00.txt, 0_01.txt, 0_02.txt, 0_03.txt, 1_00.txt, 1_01.txt, 1_02.txt, 1_03.txt, 1_04.txt, 1_05.txt, 1_06.txt, 1_07.txt, 1_08.txt, 1_09.txt, 1_10.txt, 1_11.txt, 1_12.txt, 1_13.txt, 1_14.txt, 1_15.txt, 1_16.txt, 1_17.txt, 1_18.txt, 1_19.txt, 1_20.txt, 1_21.txt, 1_22.txt, 1_23.txt, 1_24.txt, 1_25.txt, 1_26.txt, 1_27.txt, 1_28.txt
Case Name Status Exec Time Memory
0_00.txt AC 1 ms 256 KB
0_01.txt AC 1 ms 256 KB
0_02.txt AC 1 ms 384 KB
0_03.txt AC 1 ms 384 KB
1_00.txt AC 1 ms 256 KB
1_01.txt AC 9 ms 8576 KB
1_02.txt AC 18 ms 8576 KB
1_03.txt AC 3 ms 7552 KB
1_04.txt AC 3 ms 7552 KB
1_05.txt AC 3 ms 7552 KB
1_06.txt AC 4 ms 7552 KB
1_07.txt AC 4 ms 7552 KB
1_08.txt AC 4 ms 7552 KB
1_09.txt AC 5 ms 7552 KB
1_10.txt AC 5 ms 7680 KB
1_11.txt AC 7 ms 7680 KB
1_12.txt AC 5 ms 7808 KB
1_13.txt AC 11 ms 8064 KB
1_14.txt AC 7 ms 8192 KB
1_15.txt AC 8 ms 8320 KB
1_16.txt AC 9 ms 8448 KB
1_17.txt AC 4 ms 7552 KB
1_18.txt AC 3 ms 7552 KB
1_19.txt AC 4 ms 7552 KB
1_20.txt AC 3 ms 7552 KB
1_21.txt AC 3 ms 7552 KB
1_22.txt AC 3 ms 7552 KB
1_23.txt AC 3 ms 7552 KB
1_24.txt AC 3 ms 7552 KB
1_25.txt AC 4 ms 7552 KB
1_26.txt AC 3 ms 7552 KB
1_27.txt AC 4 ms 7552 KB
1_28.txt AC 4 ms 7552 KB