Cod sursa(job #2243045)

Utilizator stelian2000Stelian Chichirim stelian2000 Data 19 septembrie 2018 20:44:30
Problema Salvare Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.34 kb
#include <bits/stdc++.h>

using namespace std;

int d[1010],cnt[1010],n,nr;
char vaz[1010];
vector<int> v[1010];

void dfs(int nod,int tata,int dist)
{
    for(int i=0;i<v[nod].size();i++)
        if(v[nod][i]!=tata) dfs(v[nod][i],nod,dist);
    d[nod]=n+10;cnt[nod]=dist;
    for(int i=0;i<v[nod].size();i++)
    {
        int vec=v[nod][i];
        if(vec==tata) continue;
        d[nod]=min(d[nod],d[vec]+1);
        cnt[nod]=min(cnt[nod],cnt[vec]-1);
    }
    if(d[nod]<=cnt[nod]) cnt[nod]=n+10;
    if(cnt[nod]==0)
    {
        d[nod]=0;
        cnt[nod]=n+10;
        vaz[nod]=1;
        nr++;
    }
}

int main()
{
    freopen("salvare.in","r",stdin);
    freopen("salvare.out","w",stdout);
    int k,x,y;
    scanf("%d%d",&n,&k);
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        v[x].push_back(y);
        v[y].push_back(x);
    }
    dfs(1,0,1);
    int st=0,dr=n;
    while(st<=dr)
    {
        int mid=(st+dr)/2;
        memset(vaz,0,sizeof(vaz));
        nr=0;
        dfs(1,0,mid);
        if(nr<=k) dr=mid-1;
        else st=mid+1;
    }
    printf("%d\n",st);
    memset(vaz,0,sizeof(vaz));
    nr=0;
    dfs(1,0,st);
    for(int i=1;i<=n;i++)
        if(vaz[i]==0 && nr<k) {vaz[i]=1;nr++;}
    for(int i=1;i<=n;i++)
        if(vaz[i]==1) printf("%d ",i);
    return 0;
}