Submission #960776

#	Time	Username	Problem	Language	Result	Execution time	Memory
960776		doducanh	Burza (COCI16_burza)	C++14	160 / 160	39 ms	1540 KiB

burza

#include<bits/stdc++.h>
using namespace std;
#define int long long
#define fi first
#define se second
int max_cover[(1<<20)+7];
vector<int>a[405];
vector<int>dep[405];
vector<vector<int>>cutoff;
vector<pair<int,int>>interval;
map<int,int>lost;
int n,k;
void process(int at,int prev,int depth)
{
    dep[depth].push_back(at);
    if(depth==k){
        lost[at]=lost.size();
        cutoff[at]={at};
        return;
    }
    for(int v:a[at])if(v!=prev){
        process(v,at,depth+1);
        cutoff[at].insert(cutoff[at].end(),cutoff[v].begin(),cutoff[v].end());
    }
}
main()
{
    ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0);
    cin>>n>>k;
    cutoff.resize(n+1);
    for(int i=1;i<=n-1;i++){
        int x,y;
        cin>>x>>y;
        x--,y--;
        a[x].push_back(y);
        a[y].push_back(x);
    }
    if(k*k>=n){
        cout<<"DA";
        return 0;
    }
    process(0,0,0);
    for(const vector<int>&cc:cutoff){
        if(cc.empty()){
            interval.push_back({-1,-1});
        }
        else interval.push_back({lost[cc.front()]+1,lost[cc.back()]+1});
    }
    max_cover[0]=0;
    for(int mask=1;mask<(1<<k);mask++){
        int &curr=max_cover[mask];
        for(int to_add=0;to_add<k;to_add++){
            if(((mask>>to_add)&1)){
                int prev=max_cover[mask^(1<<to_add)];
                for(int v:dep[to_add+1]){
                    if(prev+1>=interval[v].fi){
                        curr=max(curr,interval[v].se);
                    }
                }
            }
        }
        if(curr==lost.size()){
            cout<<"DA";
            return 0;
        }
    }
    cout<<"NE";
}

Compilation message (stderr)

burza.cpp:26:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   26 | main()
      | ^~~~
burza.cpp: In function 'int main()':
burza.cpp:62:16: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::map<long long int, long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   62 |         if(curr==lost.size()){
      |            ~~~~^~~~~~~~~~~~~

• 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