oj.uz

Submission #568283

# Submission time Handle Problem Language Result Execution time Memory
568283 2022-05-25T05:47:42 Z abc864197532 Chameleon's Love (JOI20_chameleon) C++17
4 / 100
 29 ms 572 KB 
#include <bits/stdc++.h>
using namespace std;
#define lli long long int
#define mp make_pair
#define pb push_back
#define eb emplace_back
#define pii pair <int, int>
#define X first
#define Y second
#define all(x) x.begin(), x.end()
#define rall(x) x.rbegin(), x.rend()
void abc() {cout << endl;}
template <typename T, typename ...U> void abc(T a, U ...b) {
    cout << a << ' ', abc(b...);
}
template <typename T> void printv(T l, T r) {
    for (; l != r; ++l) cout << *l << " \n"[l + 1 == r];
}
template <typename A, typename B> istream& operator >> (istream& o, pair<A, B> &a) {
    return o >> a.X >> a.Y;
}
template <typename A, typename B> ostream& operator << (ostream& o, pair<A, B> a) {
    return o << '(' << a.X << ", " << a.Y << ')';
}
template <typename T> ostream& operator << (ostream& o, vector<T> a) {
    bool is = false;
    if (a.empty()) return o << "{}";
    for (T i : a) {o << (is ? ' ' : '{'), is = true, o << i;}
    return o << '}';
}
template <typename T> struct vv : vector <vector <T>> {
    vv(int n, int m, T v) : vector <vector <T>> (n, vector <T>(m, v)) {}
    vv() {}
};
template <typename T> struct vvv : vector <vv <T>> {
    vvv(int n, int m, int k, T v) : vector <vv <T>> (n, vv <T>(m, k, v)) {}
    vvv() {}
};
#ifdef Doludu
#define test(args...) abc("[" + string(#args) + "]", args)
#define owo freopen("input.txt", "r", stdin), freopen("output.txt", "w", stdout);
#include "grader.cpp"
#else
#define test(args...) void(0)
#define owo ios::sync_with_stdio(false); cin.tie(0)
#include "chameleon.h"
#endif
const int mod = 1e9 + 7, N = 100000, logN = 20, K = 48763;

void Solve(int n) {
    vector <vector <int>> g(n * 2 + 1), adj(n * 2 + 1);
    vector <int> left, right;
    int now = 0;
    for (int i = 1; i <= 2 * n; ++i) {
        left.pb(i);
        int nxt = Query(left);
        left.pop_back();
        if (nxt > now)
            left.pb(i);
        else
            right.pb(i);
        now = nxt;
    }
    test(left, right);
    assert(left.size() == n);
    for (int i = 0; i < n; ++i) {
        int l = 0;
        while (true) {
            int r = n;
            {
                vector <int> now;
                for (int i = l; i < r; ++i) 
                    now.pb(right[i]);
                int x = Query(now);
                now.pb(left[i]);
                int y = Query(now);
                if (y > r - l)
                    break;
            }
            while (r - l > 1) {
                int m = l + r >> 1;
                vector <int> now;
                for (int i = l; i < m; ++i)
                    now.pb(right[i]);
                now.pb(left[i]);
                int y = Query(now);
                if (y <= m - l)
                    r = m;
                else
                    l = m;
            }
            test(left[i], right[l]);
            g[left[i]].pb(right[l]), g[right[l]].pb(left[i]), l++;
        }
    }
    vector <int> pr(n * 2 + 1);
    for (int i = 1; i <= n * 2; ++i) {
        if (g[i].size() == 2) {
            adj[i] = g[i];
        } else if (g[i].size() == 3) {
            if (Query({i, g[i][0], g[i][1]}) == 1)
                adj[i].pb(g[i][2]);
            else if (Query({i, g[i][0], g[i][2]}) == 1)
                adj[i].pb(g[i][1]);
            else 
                adj[i].pb(g[i][0]);
        } else {
            pr[i] = g[i][0];
        }
    }
    vector <bool> vis_dfs(n * 2 + 1, false);
    vector <int> cyc;
    function <void(int)> dfs = [&](int v) {
        vis_dfs[v] = true;
        cyc.pb(v);
        for (int u : adj[v]) if (!vis_dfs[u])
            dfs(u);
    };
    vector <bool> vis(n * 2 + 1, false);
    for (int i = 1; i <= n * 2; ++i) if (!vis_dfs[i]) {
        int st = i;
        cyc.clear(), dfs(st);
        // check pair in the same cycle
        int m = cyc.size();
        for (int j = 0; j < m; ++j) if (g[cyc[j]].size() == 3) {
            vector <int> now = g[cyc[j]];
            {
                auto it = find(all(now), cyc[(j + 1) % m]);
                assert(it != now.end());
                now.erase(it);
            }
            {
                auto it = find(all(now), cyc[(j + m - 1) % m]);
                assert(it != now.end());
                now.erase(it);
            }
            pr[cyc[j]] = now[0];
        }
        vector <int> p;
        for (int i = 1; i <= n * 2; ++i) if (!count(all(cyc), i))
            p.pb(i);
        for (int st : cyc) if (pr[st] == -1) {
            int l = 0, r = p.size();
            while (r - l > 1) {
                int m = l + r >> 1;
                vector <int> now(p.begin() + l, p.begin() + m);
                int x = Query(now);
                now.pb(st);
                int y = Query(now);
                if (y == x)
                    r = m;
                else
                    l = m;
            }
            pr[st] = p[l];
        }
    }
    for (int i = 1; i <= 2 * n; ++i) {
        assert(pr[pr[i]] == i);
        if (i < pr[i])
            Answer(i, pr[i]);
    }
}

Compilation message

In file included from /usr/include/c++/10/cassert:44,
                 from /usr/include/x86_64-linux-gnu/c++/10/bits/stdc++.h:33,
                 from chameleon.cpp:2:
chameleon.cpp: In function 'void Solve(int)':
chameleon.cpp:66:24: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   66 |     assert(left.size() == n);
      |            ~~~~~~~~~~~~^~~~
chameleon.cpp:75:21: warning: unused variable 'x' [-Wunused-variable]
   75 |                 int x = Query(now);
      |                     ^
chameleon.cpp:82:27: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   82 |                 int m = l + r >> 1;
      |                         ~~^~~
chameleon.cpp:146:27: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
  146 |                 int m = l + r >> 1;
      |                         ~~^~~

Subtask #1
4.0 / 4.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 208 KB Output is correct
2 Correct 1 ms 208 KB Output is correct
3 Correct 27 ms 400 KB Output is correct
4 Correct 24 ms 380 KB Output is correct
5 Correct 24 ms 376 KB Output is correct
6 Correct 25 ms 372 KB Output is correct
7 Correct 29 ms 392 KB Output is correct
8 Correct 25 ms 372 KB Output is correct
9 Correct 24 ms 408 KB Output is correct

Subtask #2
0 / 20.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 208 KB Output is correct
2 Correct 0 ms 208 KB Output is correct
3 Runtime error 1 ms 464 KB Execution killed with signal 6
4 Halted 0 ms 0 KB -

Subtask #3
0 / 20.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 208 KB Output is correct
2 Correct 0 ms 208 KB Output is correct
3 Runtime error 1 ms 464 KB Execution killed with signal 6
4 Halted 0 ms 0 KB -

Subtask #4
0 / 20.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 208 KB Output is correct
2 Correct 0 ms 208 KB Output is correct
3 Runtime error 5 ms 572 KB Execution killed with signal 6
4 Halted 0 ms 0 KB -

Subtask #5
0 / 36.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 208 KB Output is correct
2 Correct 1 ms 208 KB Output is correct
3 Correct 27 ms 400 KB Output is correct
4 Correct 24 ms 380 KB Output is correct
5 Correct 24 ms 376 KB Output is correct
6 Correct 25 ms 372 KB Output is correct
7 Correct 29 ms 392 KB Output is correct
8 Correct 25 ms 372 KB Output is correct
9 Correct 24 ms 408 KB Output is correct
10 Correct 0 ms 208 KB Output is correct
11 Correct 0 ms 208 KB Output is correct
12 Runtime error 1 ms 464 KB Execution killed with signal 6
13 Halted 0 ms 0 KB -