제출 #249361

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
249361		SamAnd	코알라 (APIO17_koala)	C++17	73 / 100	70 ms	1500 KiB

koala

#include "koala.h"
#include <bits/stdc++.h>
using namespace std;
#define m_p make_pair
#define all(x) (x).begin(),(x).end()
#define sz(x) ((int)(x).size())
#define fi first
#define se second
const int N = 102;

int n, w;

int minValue(int N, int W)
{
    // TODO: Implement Subtask 1 solution here.
    // You may leave this function unmodified if you are not attempting this
    // subtask.
    n = N;
    w = W;

    int* a;
    a = new int[n];
    for (int i = 0; i < n; ++i)
        a[i] = 0;

    a[0] = 1;

    int* b = new int[n];
    playRound(a, b);

    for (int i = 0; i < n; ++i)
    {
        if (b[i] == 0)
            return i;
    }
    return 0;
}

int maxValue(int N, int W)
{
    // TODO: Implement Subtask 2 solution here.
    // You may leave this function unmodified if you are not attempting this
    // subtask.
    n = N;
    w = W;

    vector<int> v;
    for (int i = 0; i < n; ++i)
        v.push_back(i);

    while (1)
    {
        if (sz(v) == 1)
            return v[0];

        int* a;
        a = new int[n];
        for (int i = 0; i < n; ++i)
            a[i] = 0;

        for (int i = 0; i < v.size(); ++i)
        {
            a[v[i]] = W / sz(v);
        }

        int* b;
        b = new int[n];
        playRound(a, b);

        vector<int> nv;
        for (int i = 0; i < v.size(); ++i)
        {
            if (b[v[i]] > a[v[i]])
                nv.push_back(v[i]);
        }
        v = nv;
    }
    return 0;
}

bool c[N];
vector<int> g[N];

bool dfs(int x, int y)
{
    if (c[x])
        return false;
    c[x] = true;
    if (x == y)
        return true;
    for (int i = 0; i < g[x].size(); ++i)
    {
        int h = g[x][i];
        if (dfs(h, y))
            return true;
    }
    return false;
}

bool so(int i, int j)
{
    if (i == j)
        return false;

    memset(c, false, sizeof c);
    if (dfs(i, j))
        return false;
    memset(c, false, sizeof c);
    if (dfs(j, i))
        return true;

    int l = 1;
    int r = 14;

    while (1)
    {
        int m = (l + r) / 2;

        int* a;
        a = new int[n];
        for (int i = 0; i < n; ++i)
            a[i] = 0;
        a[i] = a[j] = m;

        if (m * 2 > w)
        {
            r = m - 1;
            continue;
        }

        int* b;
        b = new int[n];
        playRound(a, b);

        if (b[i] > a[i] && b[j] <= a[j])
        {
            g[i].push_back(j);
            return false;
        }
        if (b[i] <= a[i] && b[j] > a[j])
        {
            g[j].push_back(i);
            return true;
        }

        if (b[i] <= a[i])
            r = m - 1;
        else
            l = m + 1;
    }
    assert(false);
}

int greaterValue(int N, int W)
{
    // TODO: Implement Subtask 3 solution here.
    // You may leave this function unmodified if you are not attempting this
    // subtask.
    for (int i = 0; i < n; ++i)
        g[i].clear();
    n = N;
    w = W;

    if (so(0, 1))
        return 1;
    return 0;
}

void allValues(int N, int W, int *P)
{
    for (int i = 0; i < n; ++i)
        g[i].clear();
    n = N;
    w = W;

    if (W == 2 * N)
    {
        // TODO: Implement Subtask 4 solution here.
        // You may leave this block unmodified if you are not attempting this
        // subtask.
        for (int i = 0; i < n; ++i)
            P[i] = 0;
        for (int x = n; x >= 1; --x)
        {
            vector<int> v;
            for (int i = 0; i < n; ++i)
            {
                if (P[i])
                    continue;
                v.push_back(i);
            }

            while (1)
            {
                if (sz(v) == 1)
                {
                    P[v[0]] = x;
                    break;
                }

                int* a;
                a = new int[n];
                for (int i = 0; i < n; ++i)
                    a[i] = 0;

                for (int i = 0; i < v.size(); ++i)
                {
                    a[v[i]] = W / sz(v);
                }

                int* b;
                b = new int[n];
                playRound(a, b);

                vector<int> nv;
                for (int i = 0; i < v.size(); ++i)
                {
                    if (b[v[i]] > a[v[i]])
                        nv.push_back(v[i]);
                }
                v = nv;
            }
        }
    }
    else
    {
        // TODO: Implement Subtask 5 solution here.
        // You may leave this block unmodified if you are not attempting this
        // subtask.
        int* a;
        a = new int[n];
        for (int i = 0; i < n; ++i)
            a[i] = i;
        sort(a, a + n, so);
        for (int i = 0; i < n; ++i)
            P[a[i]] = i + 1;
    }
}

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

koala.cpp: In function 'int maxValue(int, int)':
koala.cpp:61:27: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
         for (int i = 0; i < v.size(); ++i)
                         ~~^~~~~~~~~~
koala.cpp:71:27: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
         for (int i = 0; i < v.size(); ++i)
                         ~~^~~~~~~~~~
koala.cpp: In function 'bool dfs(int, int)':
koala.cpp:91:23: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for (int i = 0; i < g[x].size(); ++i)
                     ~~^~~~~~~~~~~~~
koala.cpp: In function 'void allValues(int, int, int*)':
koala.cpp:206:35: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
                 for (int i = 0; i < v.size(); ++i)
                                 ~~^~~~~~~~~~
koala.cpp:216:35: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
                 for (int i = 0; i < v.size(); ++i)
                                 ~~^~~~~~~~~~

• Subtask 1: 4 / 4
• Subtask 2: 15 / 15
• Subtask 3: 18 / 18
• Subtask 4: 10 / 10
• Subtask 5: 26 / 53