Submission #577278

#	Time	Username	Problem	Language	Result	Execution time	Memory
577278		keta_tsimakuridze	Burza (COCI16_burza)	C++14	160 / 160	127 ms	9996 KiB

burza

#include<bits/stdc++.h>
using namespace std;
#define pii pair<int,int>
#define f first
#define s second
#define endl "\n"
const int N = 403, mod = 1e9 + 7; //!
int t, dp[N][1 << 20], h[N], tmin[N], timer, tmout[N], p[N];
vector<int> v, V[N];
int n, k;
void dfs(int u) {
	h[u] = h[p[u]] + 1;
	if(h[u] == k) {
		v.push_back(u); 
		tmin[u] = tmout[u] = ++timer;
		return;
	}
	tmin[u] = n + 5;
	for(int i = 0; i < V[u].size(); i++) {
		if(V[u][i] == p[u]) continue;
		p[V[u][i]] = u;
		dfs(V[u][i]);
		tmin[u] = min(tmin[u], tmin[V[u][i]]);
	}
	tmout[u] = timer;
}
main() {
	ios_base::sync_with_stdio(false),cin.tie(0),cout.tie(0);
	cin >> n >> k;
	++k;
	if(k > 20) {
		cout << "DA";
		return 0;
	}
	for(int i = 2; i <= n; i++) {
		int u, v;
		cin >> u >> v;
		V[u].push_back(v);
		V[v].push_back(u);
	}
	dfs(1); 
	dp[1][0] = 1;
	for(int i = 0; i < v.size(); i++) {

		int x = v[i]; 
		while(x != 1) {
			if(tmin[x] != tmin[v[i]]) break;
			for(int j = 0; j < (1 << k); j++) {
				if(j & (1 << (h[x] - 2)) || !dp[tmin[x]][j]) continue;
				dp[tmout[x] + 1][j ^ (1 << (h[x] - 2))] = 1; 
			}
			x = p[x];
		}
	}
	int ans = 0;
	for(int j = 0; j < (1 << k); j++) ans |= dp[timer + 1][j];
	cout << (ans ? "DA" : "NE");
	
}

Compilation message (stderr)

burza.cpp: In function 'void dfs(int)':
burza.cpp:19:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   19 |  for(int i = 0; i < V[u].size(); i++) {
      |                 ~~^~~~~~~~~~~~~
burza.cpp: At global scope:
burza.cpp:27:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   27 | main() {
      | ^~~~
burza.cpp: In function 'int main()':
burza.cpp:43:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   43 |  for(int i = 0; i < v.size(); i++) {
      |                 ~~^~~~~~~~~~

• Subtask 1: 16 / 16
• Subtask 2: 16 / 16
• Subtask 3: 16 / 16
• Subtask 4: 16 / 16
• Subtask 5: 16 / 16
• Subtask 6: 16 / 16
• Subtask 7: 16 / 16
• Subtask 8: 16 / 16
• Subtask 9: 16 / 16
• Subtask 10: 16 / 16