Submission #1175857

#	Time	Username	Problem	Language	Result	Execution time	Memory
1175857		InvMOD	Burza (COCI16_burza)	C++20	0 / 160	0 ms	400 KiB

burza

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

#define fi first
#define se second
#define gcd __gcd
#define sz(v) (int) v.size()
#define pb push_back
#define pi pair<int,int>
#define all(v) (v).begin(), (v).end()
#define compact(v) (v).erase(unique(all(v)), (v).end())
#define FOR(i, a, b) for(int i = (a); i <= (b); i++)
#define REV(i, a, b) for(int i = (a); i >= (b); i--)
#define dbg(x) "[" #x " = " << (x) << "]"
///#define int long long

using ll = long long;
using ld = long double;
using ull = unsigned long long;

template<typename T> bool ckmx(T& a, const T& b){if(a < b) return a = b, true; return false;}
template<typename T> bool ckmn(T& a, const T& b){if(a > b) return a = b, true; return false;}

const int N = 2e5+5;
const ll MOD = 1e9+7;
const ll INF = 1e18;




void solve()
{
    int n,k; cin >> n >> k;

    vector<vector<int>> adj(n + 1, vector<int>());
    for(int i = 0; i < n - 1; i++){
        int u,v; cin >> u >> v;
        adj[u].pb(v), adj[v].pb(u);
    }

    vector<int> h(n + 1);
    function<void(int,int)> getH = [&](int x, int p) -> void{
        for(int v : adj[x])if(v != p){
            h[v] = h[x] + 1;
            getH(v, x);
        }
    };
    getH(1, -1);

    vector<int> loose(n + 1);
    function<bool(int,int)> check = [&](int x, int p) -> bool{
        if(h[x] >= k){
            loose[x] = 1;
            return false;
        }

        for(int v : adj[x])if(v != p){
            if(check(v, x)) return true;

            loose[x] += loose[v];
        }

        //cout << x <<" " << loose[x] << "\n";

        if(k - h[x] >= loose[x]) return true;
        return false;
    };

    cout << (check(1, -1) ? "DA" : "NE") << "\n";
}

signed main()
{
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);

    #define name "InvMOD"
    if(fopen(name".INP", "r")){
        freopen(name".INP","r",stdin);
        freopen(name".OUT","w",stdout);
    }

    int t = 1; //cin >> t;
    while(t--) solve();
    return 0;
}

Compilation message (stderr)

burza.cpp: In function 'int main()':
burza.cpp:80:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   80 |         freopen(name".INP","r",stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~
burza.cpp:81:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   81 |         freopen(name".OUT","w",stdout);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~

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