제출 #202475

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
202475		rdd6584	Drvca (COCI19_drvca)	C++14	0 / 110	107 ms	9160 KiB

drvca

#include <bits/stdc++.h>
#define sz(x) (int)(x).size()
#define all(x) (x).begin(), (x).end()
using namespace std;

typedef long long ll;
typedef unsigned long long llu;
typedef pair<int, int> pii;
typedef pair<int, pii> piii;
typedef pair<ll, ll> pll;
typedef pair<ll, int> pli;
typedef pair<int, ll> pil;
typedef pair<string, int> psi;
typedef pair<char, int> pci;
typedef pair<int, char> pic;
const ll MOD = (ll)1e9 + 7;
const long double PI = 3.141592653589793238462643383279502884197;

priority_queue<int, vector<int>, greater<int>> pq;

vector<int> v;
multiset<int> se, tmp;

int vec[100000];
int n;

void print() {
    printf("%d\n", sz(v));
    for (int i = 0; i < sz(v); i++)
        printf("%d ", v[i]);
    printf("\n%d\n", sz(tmp));
    for (int i : tmp)
        printf("%d ", i);
    exit(0);
}
void check() {
    if (sz(v) == n) {
        se.insert(v.back());
        v.pop_back();
    }

    tmp = se;
    if (sz(se) == 1) print();
    int kk = *(++se.begin());
    int d = kk - *se.begin();
    int next;

    se.erase(se.begin());
    next = *se.begin() + d;
    se.erase(se.begin());

    int flag = 0;
    while (!se.empty()) {
        auto t = se.find(next);
        if (t == se.end()) {
            if (!flag && next == v.back()) {
                flag++;
                v.pop_back();
                se.insert(next);
            }
            else break;
        }
        se.erase(t);
        next += d;
    }

    if (se.empty()) print();
    v.clear();
    se.clear();
}

int main() {
    scanf("%d", &n);

    for (int i = 0; i < n; i++)
        scanf("%d", &vec[i]);
    sort(vec, vec + n);

    if (n == 2) { return !printf("1\n%d\n1\n%d", vec[0], vec[1]); }

    // 1~2 등차 해보고 안되면..
    int d = 0;
    v.push_back(vec[0]);
    v.push_back(vec[1]);
    d = vec[1] - vec[0];

    for (int i = 2; i < n; i++) se.insert(vec[i]);
    int next = vec[1] + d;

    while (!se.empty()) {
        auto t = se.find(next);
        if (t == se.end()) break;
        v.push_back(*t);
        se.erase(t);
        next += d;
    }
    check();


    v.push_back(vec[0]);
    v.push_back(vec[2]);
    d = vec[2] - vec[0];

    se.insert(vec[1]);
    for (int i = 3; i < n; i++) se.insert(vec[i]);
    next = vec[2] + d;

    while (!se.empty()) {
        auto t = se.find(next);
        if (t == se.end()) break;
        v.push_back(*t);
        se.erase(t);
        next += d;
    }
    check();


    v.push_back(vec[1]);
    v.push_back(vec[2]);
    d = vec[2] - vec[1];

    se.insert(vec[0]);
    for (int i = 3; i < n; i++) se.insert(vec[i]);
    next = vec[2] + d;

    while (!se.empty()) {
        auto t = se.find(next);
        if (t == se.end()) break;
        v.push_back(*t);
        se.erase(t);
        next += d;
    }
    check();

    printf("-1");
}

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

drvca.cpp: In function 'int main()':
drvca.cpp:73:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     scanf("%d", &n);
     ~~~~~^~~~~~~~~~
drvca.cpp:76:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
         scanf("%d", &vec[i]);
         ~~~~~^~~~~~~~~~~~~~~

• Subtask 1: 0 / 20
• Subtask 2: 0 / 30
• Subtask 3: 0 / 30
• Subtask 4: 0 / 30