Submission #133842

#	Time	Username	Problem	Language	Result	Execution time	Memory
133842		forelax	Burza (COCI16_burza)	C++14	0 / 160	2 ms	376 KiB

burza

#include<bits/stdc++.h>
using namespace std;
vector<vector<int> > ng;
vector<int> dp;
vector<bool> md;
int n,k;
void build(int ind,int par,int dep){
    if(dep==k){
        dp[ind]=1e6;
        md[ind]=true;
        return;
    }
    int mx=0;
    int sm=0;
    for(int i = 0 ; i < ng[ind].size() ; i ++){
        int ne=ng[ind][i];
        if(ne==par)continue;
        build(ne,ind,dep+1);
        if(md[ne]){
            md[ind]=true;
            sm+=dp[ind];
            mx=max(dp[ind],mx);
        }
    }
    dp[ind]=sm-mx+min(mx,1);
    if(dp[ind]>=k-dep)
        dp[ind]=1e6;
}
int main(){
    cin>>n>>k;
    ng.resize(n);
    for(int i = 0,a,b ; i < n-1 ; i ++){
        cin>>a>>b;a--;b--;
        ng[a].push_back(b);
        ng[b].push_back(a);
    }
    dp.resize(n);
    md.resize(n);
    build(0,-1,0);
//    for(int i = 0 ; i < n ; i ++)
//        cout<<i<<" "<<dp[i]<<endl;;
//    cout<<"NE";
//    return 0;
    if(dp[0]<k){
        cout<<"DA";
    }else{
        cout<<"NE";
    }
}

Compilation message (stderr)

burza.cpp: In function 'void build(int, int, int)':
burza.cpp:15:23: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(int i = 0 ; i < ng[ind].size() ; i ++){
                     ~~^~~~~~~~~~~~~~~~

• 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