제출 #608599

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
608599		Hanksburger	Split the Attractions (IOI19_split)	C++17	40 / 100	133 ms	18108 KiB

split

#include "split.h"
#include <bits/stdc++.h>
using namespace std;
int sz[100005], x[100005], node, cnt;
vector<int> adj[100005], ans;
bool visited[100005];
void dfs(int u)
{
    visited[u]=1;
    sz[u]=1;
    for (int v:adj[u])
    {
        if (!visited[v])
        {
            dfs(v);
            sz[u]+=sz[v];
        }
    }
    x[sz[u]]=u;
}
void dfs2(int u, bool b)
{
    visited[u]=1;
    if (b)
    {
        ans[u]=1;
        cnt--;
    }
    if (cnt)
    {
        for (int v:adj[u])
        {
            if (!visited[v])
            {
                dfs2(v, b|(v==node));
                if (!cnt)
                    break;
            }
        }
    }
}
void dfs3(int u)
{
    visited[u]=1;
    ans[u]=2;
    cnt--;
    if (cnt)
    {
        for (int v:adj[u])
        {
            if (!visited[v] && ans[v]==3)
            {
                dfs3(v);
                if (!cnt)
                    break;
            }
        }
    }
}
vector<int> find_split(int n, int a, int b, int c, vector<int> p, vector<int> q)
{
    int swapped=0;
    if (a>=b && a>=c)
    {
        swap(a, c);
        swapped=1;
    }
    else if (b>=a && b>=c)
    {
        swap(b, c);
        swapped=2;
    }
    for (int i=0; i<p.size(); i++)
    {
        adj[p[i]].push_back(q[i]);
        adj[q[i]].push_back(p[i]);
    }
    for (int z=0; z<2; z++)
    {
        for (int i=1; i<=n; i++)
            x[i]=-1;
        for (int i=0; i<n; i++)
            visited[i]=0;
        dfs(0);
        int ind=a;
        while (x[ind]==-1)
            ind++;
        if (ind+b<=n)
        {
            node=x[ind];
            for (int i=0; i<n; i++)
                ans.push_back(3);
            cnt=a;
            for (int i=0; i<n; i++)
                visited[i]=0;
            dfs2(0, 0);
            cnt=b;
            for (int i=0; i<n; i++)
                visited[i]=0;
            dfs3(0);
            if (z)
            {
                for (int i=0; i<n; i++)
                    ans[i]=(ans[i]==3)?3:(3-ans[i]);
            }
            if (swapped==1)
            {
                for (int i=0; i<n; i++)
                    ans[i]=4-ans[i];
            }
            else if (swapped==2)
            {
                for (int i=0; i<n; i++)
                    ans[i]=(ans[i]==1)?1:(5-ans[i]);
            }
            return ans;
        }
        swap(a, b);
    }
    for (int i=0; i<n; i++)
        ans.push_back(0);
    return ans;
}

컴파일 시 표준 에러 (stderr) 메시지

split.cpp: In function 'std::vector<int> find_split(int, int, int, int, std::vector<int>, std::vector<int>)':
split.cpp:73:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   73 |     for (int i=0; i<p.size(); i++)
      |                   ~^~~~~~~~~

• Subtask 1: 7 / 7
• Subtask 2: 11 / 11
• Subtask 3: 22 / 22
• Subtask 4: 0 / 24
• Subtask 5: 0 / 36