oj.uz

답안 #544146

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
544146 2022-04-01T08:30:15 Z levladiator Pastiri (COI20_pastiri) C++14
18 / 100
 627 ms 65612 KB 
#include <bits/stdc++.h>
#define pb push_back
#define pf push_front
#define ll long long
#define ull unsigned long long
#define pii pair<int,int>
#define pll pair<ll,ll>
#define EPSILON 0.000001
#define CH (*s-'a')

using namespace std;

const ll NMAX = 5e5 + 5, INF = 1e18, MOD = 1e9 + 7, MMAX = 1e2 + 5, inf = 1e9;

ifstream fin("sufle.in");
ofstream fout("sufle.out");

int N,K;
int sheep[NMAX],dep[NMAX],dist[NMAX],anc[NMAX];
bool vis[NMAX];
vector<int> adj[NMAX],ans;

struct nod
{
    int dep,node,anc;
};
nod v[NMAX];

bool cmp(nod a,nod b)
{
    return a.dep>b.dep;
}

void bfs()
{
    queue<int> Q;
    for(int i=1;i<=N;i++)
    {
        if(sheep[i])Q.push(i);
    }
    while(!Q.empty())
    {
        int node=Q.front();
        Q.pop();
        for(auto vec : adj[node])
        {
            if(!sheep[vec]&&!dist[vec])
            {
                dist[vec]=dist[node]+1;
                Q.push(vec);
            }
        }
    }
}

void dfs(int node,int prev,int last)
{
    dep[node]=dep[prev]+1;
    if(sheep[node])anc[node]=last;
    if(dist[last]+dep[last]<dist[node]+dep[node])
    {
        last=node;
    }
    for(auto son : adj[node])
    {
        if(son==prev)
            continue;

        dfs(son,node,last);
    }
}

void mark(int node)
{
    vis[node]=1;
    for(auto vec : adj[node])
    {
        if(vis[vec])
            continue;

        if(dist[vec]+1==dist[node])
        {
            mark(vec);
        }
    }
}

int main()
{
    cin.tie ( 0 )->sync_with_stdio ( 0 );
    cin.tie ( NULL );
    cout.tie ( NULL );

    cin>>N>>K;
    for(int i=1;i<N;i++)
    {
        int a,b;
        cin>>a>>b;
        adj[a].pb(b);
        adj[b].pb(a);
    }
    for(int i=1;i<=K;i++)
    {
        int x;
        cin>>x;
        sheep[x]=1;
    }
    bfs();
    dfs(1,0,1);
    int cnt=0;
    for(int i=1;i<=N;i++)
    {
        if(sheep[i])
        {
            cnt++;
            v[cnt].node=i;
            v[cnt].dep=dep[i];
            v[cnt].anc=anc[i];
        }
    }
    sort(v+1,v+K+1,cmp);
    for(int i=1;i<=K;i++)
    {
        int node=v[i].node;
        if(!vis[node])
        {
            ans.pb(anc[node]);
            mark(anc[node]);
        }
    }
    cout<<ans.size()<<'\n';
    for(auto x : ans)
    {
        cout<<x<<' ';
    }

    return 0;
}

Subtask #1
0 / 8.0

# 결과 실행 시간 메모리 Grader output
1 Correct 138 ms 55372 KB Output is correct
2 Correct 193 ms 58852 KB Output is correct
3 Correct 189 ms 58780 KB Output is correct
4 Incorrect 211 ms 65612 KB Sheep 4 not protected
5 Halted 0 ms 0 KB -

Subtask #2
18.0 / 18.0

# 결과 실행 시간 메모리 Grader output
1 Correct 8 ms 12244 KB Output is correct
2 Correct 8 ms 12348 KB Output is correct
3 Correct 377 ms 32788 KB Output is correct
4 Correct 375 ms 35092 KB Output is correct
5 Correct 569 ms 32468 KB Output is correct
6 Correct 627 ms 34504 KB Output is correct

Subtask #3
0 / 23.0

# 결과 실행 시간 메모리 Grader output
1 Correct 7 ms 12116 KB Output is correct
2 Incorrect 7 ms 12212 KB Sheep 63 not protected
3 Halted 0 ms 0 KB -

Subtask #4
0 / 51.0

# 결과 실행 시간 메모리 Grader output
1 Correct 507 ms 37200 KB Output is correct
2 Incorrect 492 ms 37116 KB Sheep 438733 not protected
3 Halted 0 ms 0 KB -