CSES - Datatähti Open 2021 - Results
Submission details
Task:Sorting
Sender:meatrow
Submission time:2021-01-29 21:51:25 +0200
Language:C++ (C++17)
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
Test results
testverdicttimegroup
#1ACCEPTED0.01 s1, 2details
#2ACCEPTED0.66 s2details
#3ACCEPTED0.10 s1, 2details
#4--1, 2details

Compiler report

input/code.cpp: In member function 'void Fenwick::add(int)':
input/code.cpp:33:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
         for (; pos < data.size(); pos |= pos + 1) {
                ~~~~^~~~~~~~~~~~~

Code

#pragma GCC optimize("O3")
#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx")
#include <bits/stdc++.h>
 
using namespace std;
 
using ll = long long;
using ld = long double;
 
const int mod = 1e9 + 7;
 
ll binpow(ll a, ll p) {
    ll res = 1;
    while (p) {
        if (p & 1) {
            (res *= a) %= mod;
        }
        p >>= 1;
        (a *= a) %= mod;
    }
    return res;
}

class Fenwick {
private:
    vector<int> data;
public:
    Fenwick(int n) {
        data.assign(n, 0);
    }

    void add(int pos) {
        for (; pos < data.size(); pos |= pos + 1) {
            data[pos]++;
        }
    }
    int get(int pos) {
        int res = 0;
        for (; pos >= 0; pos = (pos & (pos + 1)) - 1) {
            res += data[pos];
        }
        return res;
    }
};

void solve() {
    int n;
    cin >> n;
    Fenwick f(n + 1);
    vector<int> a(n);
    int inv = 0;
    for (int i = 0; i < n; i++) {
        cin >> a[i];
        inv += f.get(n) - f.get(a[i]);
        f.add(a[i]);
    }
    if (inv % 2) {
        cout << "NO\n";
        return;
    }

    if (n > 8) {
        cout << "YES\n";
        return;
    }
    map<vector<int>, int> mp;
    mp[a] = 1;
    queue<vector<int>> q;
    q.push(a);
    vector<int> p(n);
    iota(p.begin(), p.end(), 1);
    while (!q.empty()) {
        auto v = q.front();
        q.pop();
        if (v == p) {
            cout << "YES\n";
            return;
        }
        for (int i = 0; i + 1 < n; i++) {
            for (int j = i + 2; j + 1 < n; j++) {
                swap(v[i], v[j]);
                swap(v[i + 1], v[j + 1]);
                if (!mp.count(v)) {
                    mp[v] = 1;
                    q.push(v);
                }
            }
        }
    }

    cout << "NO\n";
}
 
int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    int T = 1;
    cin >> T;
    for (int tc = 1; tc <= T; tc++) {
        // cout << "Case #" << tc << ": ";
        solve();
    }
    return 0;
}

Test details

Test 1

Group: 1, 2

Verdict: ACCEPTED

input
153
1
1
2
1 2
...

correct output
YES
YES
NO
NO
NO
...

user output
YES
YES
NO
NO
NO
...
Truncated

Test 2

Group: 2

Verdict: ACCEPTED

input
1000
59
35 29 32 50 11 15 9 21 19 45 2...

correct output
YES
NO
YES
NO
YES
...

user output
YES
NO
YES
NO
YES
...
Truncated

Test 3

Group: 1, 2

Verdict: ACCEPTED

input
720
6
1 6 4 5 2 3
6
6 3 2 1 5 4
...

correct output
YES
NO
NO
NO
YES
...

user output
YES
NO
NO
NO
YES
...
Truncated

Test 4

Group: 1, 2

Verdict:

input
1000
8
7 4 2 8 6 3 5 1
8
3 8 2 7 5 4 6 1
...

correct output
NO
NO
YES
NO
YES
...

user output
(empty)