結果

問題 No.1306 Exactly 2 Digits
ユーザー hitonanode
提出日時 2020-12-03 01:32:58
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 152 ms / 2,000 ms
コード長 2,757 bytes
コンパイル時間 2,895 ms
コンパイル使用メモリ 216,228 KB
最終ジャッジ日時 2025-01-16 13:57:49
ジャッジサーバーID
(参考情報)
judge3 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 123
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;
using pint = pair<int, int>;
template <typename T1, typename T2> pair<T1, T2> operator+(const pair<T1, T2> &l, const pair<T1, T2> &r) { return make_pair(l.first + r.first, l.second + r.second); }
template <typename T1, typename T2> pair<T1, T2> operator-(const pair<T1, T2> &l, const pair<T1, T2> &r) { return make_pair(l.first - r.first, l.second - r.second); }

pint query(int i, int j) {
    cout << "? " << i + 1 << ' ' << j + 1 << endl;
    int x, y;
    cin >> x >> y;
    return make_pair(x, y);
}

pint simulate(pint pi, pint pj) {
    pint ret = pi - pj;
    if (ret.first > ret.second) swap(ret.first, ret.second);
    return ret;
}

int main()
{
    int N;
    cin >> N;
    const int L = N * N - N;

    map<int, int> freqcnt;
    freqcnt[0] += 2;
    map<pint, vector<int>> p2i;

    for (int i = 1; i < L; i++) {
        auto [x, y] = query(i, 0);
        p2i[pint(x, y)].emplace_back(i);
        if (x == 0 or y == 0) freqcnt[x + y]++;
    }

    const int lo = freqcnt.begin()->first;
    const int hi = prev(freqcnt.end())->first;
    int lo2 = lo, hi2 = hi;
    while (freqcnt[lo2] == 1) lo2++;
    while (freqcnt[hi2] == 1) hi2--;

    vector<pint> ret(L, pint(-1, -1));

    if (hi2 - lo == N - 1 and hi - lo2 == N - 2) {
        ret[0] = make_pair(-lo2 + 1, -lo); // (十, 一)
    } else if (hi2 - lo == N - 2 and hi - lo2 == N - 1) {
        ret[0] = make_pair(-lo + 1, -lo2);
    }

    auto is_valid = [&](pint p) -> bool { return p.first > 0 and p.first < N and p.second >= 0 and p.second < N; };

    vector<int> ics;
    for (const auto [p, is] : p2i) {
        pint pa = p;
        if (is.size() == 1) {
            int i = is[0];
            pint pcand = ret[0] + pa;
            if (is_valid(pcand)) ret[i] = pcand;
            else {
                swap(pa.first, pa.second);
                ret[i] = ret[0] + pa;
            }
            ics.emplace_back(i);
        }
    }
    assert(ics.size() > 0);

    for (const auto [p, is] : p2i) {
        if (is.size() == 2) {
            pint pa = p;
            const int i = is[0], j = is[1];

            const pint p1 = ret[0] + pa;
            swap(pa.first, pa.second);
            const pint p2 = ret[0] + pa;

            for (auto ic : ics) {
                pint psim1 = simulate(p1, ret[ic]);
                pint psim2 = simulate(p2, ret[ic]);
                if (psim1 != psim2) {
                    pint ptmp = query(i, ic);
                    if (psim1 == ptmp) ret[i] = p1, ret[j] = p2;
                    else ret[i] = p2, ret[j] = p1;
                    break;
                }
            }
        }
    }
    cout << "! ";
    for (auto x : ret) cout << x.first * N + x.second << ' ';
    cout << endl;
}
0