Cod sursa(job #2389978)

Utilizator Ruxandra985Nanu Ruxandra Laura Ruxandra985 Data 27 martie 2019 17:41:10
Problema Salvare Scor 50
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.49 kb
#include <cstdio>
#include <bitset>
#include <vector>
#include <algorithm>
#define DIMN 1001
using namespace std;
int elem,salv[DIMN],d[DIMN],k;
bitset <DIMN> f;
vector <int> v[DIMN];
void dfs (int nod,int x){
    int i,vecin;
    f[nod]=1;
    if (nod!=1 && v[nod].size()==1){ /// frunza
        d[nod]=x;
        return;
    }
    d[nod]=2*x+1;
    for (i=0;i<v[nod].size();i++){
        vecin=v[nod][i];
        if (f[vecin]==0){
            dfs(vecin,x);
            d[nod]=min(d[nod],d[vecin]-1);
        }
    }
    if (!d[nod]){ /// nod marcat
        d[nod]=2*x;
        salv[++elem]=nod;
    }
}
int verif (int x){
    elem=0;
    f.reset();
    dfs (1,x);
    if (d[1]<=x)
        salv[++elem]=1;
    return elem<=k;
}
int main()
{
    FILE *fin=fopen ("salvare.in","r");
    FILE *fout=fopen ("salvare.out","w");
    int n,i,x,y,st,dr,mid,add,p;
    fscanf (fin,"%d%d",&n,&k);
    for (i=1;i<n;i++){
        fscanf (fin,"%d%d",&x,&y);
        v[x].push_back(y);
        v[y].push_back(x);
    }
    st=1;
    dr=n;
    while (st<=dr){
        mid=(st+dr)/2;
        if (verif (mid))
            dr=mid-1;
        else st=mid+1;
    }
    verif (st);
    fprintf (fout,"%d\n",st);
    f.reset();
    add=k-elem;
    p=1;
    while (add){
        if (!f[p]){
            salv[++elem]=p;
            add--;
        }
        p++;
    }
    sort (salv+1,salv+elem+1);
    for (i=1;i<=elem;i++)
        fprintf (fout,"%d ",salv[i]);
    return 0;
}