Submission #23735814

---

Source Code Expand

#include<bits/stdc++.h>
using namespace std;

using ll = long long;
const int mod = 1e9+7;

vector<vector<int>> edge;
vector<vector<vector<vector<int>>>> dp;

void dfs(int now){
	dp[now].resize(2,vector<vector<int>>(2,vector<int>(2)));
	dp[now][0][0][0] = dp[now][1][1][1] = 1;
	for(auto next: edge[now]){
		if(!dp[next].empty()) continue;
		dfs(next);
		int N = dp[now].size()-1, M = dp[next].size()-1;
		vector<vector<vector<int>>> nex(N+M+1,vector<vector<int>>(2,vector<int>(2)));
		for(int i=0; i<=N; i++){
			for(int j=0; j<=M; j++){
				for(int k=0; k<2; k++){
					for(int l=0; l<2; l++){
						for(int m=0; m<2; m++){
							for(int n=0; n<2; n++){
								if(dp[now][i][k][l]==0 || dp[next][j][m][n]==0) continue;
								int sum = i+j+(!l&m)+(k&!n);
								nex[sum][k][l|m] += ll(dp[now][i][k][l])*dp[next][j][m][n]%mod;
								nex[sum][k][l|m] %= mod;
							}
						}
					}
				}
			}
		}
		dp[now] = nex;
	}
}

int main(){
	int N; cin >> N;
	edge.resize(N);
	dp.resize(N);
	for(int i=1; i<N; i++){
		int U,V; cin >> U >> V;
		edge[--U].push_back(--V);
		edge[V].push_back(U);
	}
	dfs(0);
	for(int i=0; i<=N; i++){
		int ans = 0;
		for(int j=0; j<2; j++){
			for(int k=0; k<2; k++){
				ans += dp[0][i][j][k];
				ans %= mod;
			}
		}
		cout << ans << endl;
	}
}

Submission Info

Compile Error

./Main.cpp: In function ‘void dfs(int)’:
./Main.cpp:25:24: warning: suggest parentheses around operand of ‘!’ or change ‘&’ to ‘&&’ or ‘!’ to ‘~’ [-Wparentheses]
   25 |         int sum = i+j+(!l&m)+(k&!n);
      |                        ^~

Judge Result