Submission #1184103

#TimeUsernameProblemLanguageResultExecution timeMemory
11841038pete8Star Trek (CEOI20_startrek)C++20
85 / 100
95 ms20804 KiB
#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define f first
#define s second
#define pii pair<int32_t,int32_t>
#define ppii pair<int,pii>
#define vi vector<int>
#define pb push_back
#define all(x) x.begin(),x.end()
#define rall(x) x.rbegin(),x.rend()
#define F(n) for(int i=0;i<n;i++)
#define lb lower_bound
#define ub upper_bound
#define fastio ios::sync_with_stdio(false);cin.tie(NULL);
#pragma GCC optimize ("03,unroll-lopps")
#define sz(x) (int)((x).size())
#define int long long
using namespace std;
const int mod=1e9+7,mxn=1e5,inf=1e18,minf=-1e18,lg=30,Mxn=1e9;
//#undef int
int n,k,m,d,q;
void setIO(string name){
	ios_base::sync_with_stdio(0); cin.tie(0);
	freopen((name+".in").c_str(),"r",stdin);
	freopen((name+".out").c_str(),"w",stdout);
}
int nim[mxn+10],nimu[mxn+10];
int nimr[mxn+10];

//rooted at, cur
vector<int>adj[mxn+10];

int mex(int c[2]){
	if(c[0]==0)return 0;
	return 1;
}
void dfs(int cur,int p){
	int c[2]={0,0};
	for(auto i:adj[cur]){
		if(i==p)continue;
		dfs(i,cur);
		c[!!nim[i]]++;
	}
	nim[cur]=mex(c);
}
void dfs2(int cur,int p){
	int c[2]={0,0};
	if(p!=-1)c[!!nimu[cur]]++;
	for(auto i:adj[cur])if(i!=p)c[!!nim[i]]++;
	nimr[cur]=mex(c);
	for(auto i:adj[cur])if(i!=p){
		c[!!nim[i]]--;
		nimu[i]=mex(c);
		c[!!nim[i]]++;
	}
	for(auto i:adj[cur])if(i!=p)dfs2(i,cur);
}
int C[mxn+10],Cu[mxn+10][2],Cr[mxn+10];
void dfs3(int cur,int p){
	int x=0;
	C[cur]+=(!nim[cur]);
	for(auto i:adj[cur])if(i!=p){
		dfs3(i,cur);
		x+=(!nim[i]);
	}
	if(x<=1){
		for(auto i:adj[cur])if(i!=p&&(nim[cur]^nim[i]))C[cur]+=C[i];
	}
}
void dfs4(int cur,int p){
	int x=0,sum=0;
	if(p!=-1)x+=(!nimu[cur]);
	for(auto i:adj[cur])if(i!=p){
		x+=(!nim[i]);
		Cu[cur][nim[i]]+=C[i];
	}
	Cr[cur]+=(!nimr[cur]);
	if(x<=1)Cr[cur]+=Cu[cur][!nimr[cur]];
	for(auto i:adj[cur])if(i!=p){
		x-=(!nim[i]);
		if(x<=1)Cu[i][x]+=(Cu[cur][!x]-((nim[i]!=x)?C[i]:0))+(!x);
		x+=(!nim[i]);
	}
	for(auto i:adj[cur])if(i!=p)dfs4(i,cur);
}
int cnt=0,D;
int pow(int ex){
	int ans=1,a=4;
	while(ex){
		if(ex&1)ans=(ans*a)%mod;
		a=(a*a)%mod;
		ex>>=1;
	}
	return ans;
}
int mul(int a,int b){return (a*b)%mod;}
int add(int a,int b){return (a+b+mod)%mod;}
mt19937 rng(time(NULL));
int dp[mxn+10][2];
int sum[2];
int32_t main(){
	fastio
	cin>>n>>D;
	if(n==2){
		cout<<pow(D);
		return 0;
	}
	for(int i=1;i<n;i++){
		int a,b;cin>>a>>b;
		adj[a].pb(b);
		adj[b].pb(a);
	}
	dfs(1,-1);
	dfs2(1,-1);
	dfs3(1,-1);
	dfs4(1,-1);

	for(int i=1;i<=n;i++){
		cnt+=nimr[i];
		sum[nimr[i]]=add(sum[nimr[i]],Cr[i]);
	}
	dp[0][1]=cnt;
	dp[0][0]=n-cnt;
	for(int i=1;i<D;i++){
		//0->1
		dp[i][1]=add(mul(-sum[1],dp[i-1][0]),mul(add(dp[i-1][0],dp[i-1][1]),mul(n,cnt)));
		dp[i][1]=add(dp[i][1],mul(sum[0],dp[i-1][0]));

		dp[i][0]=add(mul(-sum[0],dp[i-1][0]),mul(add(dp[i-1][0],dp[i-1][1]),mul(n,n-cnt)));
		dp[i][0]=add(dp[i][0],mul(sum[1],dp[i-1][0]));
	}
	D--;
	if(nimr[1])cout<<add(mul(-Cr[1],dp[D][0]),mul(add(dp[D][0],dp[D][1]),n));
	else cout<<mul(Cr[1],dp[D][0]);
}

Compilation message (stderr)

startrek.cpp:16:40: warning: bad option '-funroll-lopps' to pragma 'optimize' [-Wpragmas]
   16 | #pragma GCC optimize ("03,unroll-lopps")
      |                                        ^
startrek.cpp:23:23: warning: bad option '-funroll-lopps' to attribute 'optimize' [-Wattributes]
   23 | void setIO(string name){
      |                       ^
startrek.cpp:34:17: warning: bad option '-funroll-lopps' to attribute 'optimize' [-Wattributes]
   34 | int mex(int c[2]){
      |                 ^
startrek.cpp:38:23: warning: bad option '-funroll-lopps' to attribute 'optimize' [-Wattributes]
   38 | void dfs(int cur,int p){
      |                       ^
startrek.cpp:47:24: warning: bad option '-funroll-lopps' to attribute 'optimize' [-Wattributes]
   47 | void dfs2(int cur,int p){
      |                        ^
startrek.cpp:60:24: warning: bad option '-funroll-lopps' to attribute 'optimize' [-Wattributes]
   60 | void dfs3(int cur,int p){
      |                        ^
startrek.cpp:71:24: warning: bad option '-funroll-lopps' to attribute 'optimize' [-Wattributes]
   71 | void dfs4(int cur,int p){
      |                        ^
startrek.cpp:88:15: warning: bad option '-funroll-lopps' to attribute 'optimize' [-Wattributes]
   88 | int pow(int ex){
      |               ^
startrek.cpp:97:20: warning: bad option '-funroll-lopps' to attribute 'optimize' [-Wattributes]
   97 | int mul(int a,int b){return (a*b)%mod;}
      |                    ^
startrek.cpp:98:20: warning: bad option '-funroll-lopps' to attribute 'optimize' [-Wattributes]
   98 | int add(int a,int b){return (a+b+mod)%mod;}
      |                    ^
startrek.cpp:102:14: warning: bad option '-funroll-lopps' to attribute 'optimize' [-Wattributes]
  102 | int32_t main(){
      |              ^
startrek.cpp: In function 'void setIO(std::string)':
startrek.cpp:25:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   25 |         freopen((name+".in").c_str(),"r",stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
startrek.cpp:26:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   26 |         freopen((name+".out").c_str(),"w",stdout);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...