CSES - Datatähti Open 2021 - Results
Submission details
Task:Sorting
Sender:Victor
Submission time:2021-01-30 17:35:04 +0200
Language:C++17
Status:READY
Result:36
Feedback
groupverdictscore
#1ACCEPTED36
#20
Test results
testverdicttimegroup
#1ACCEPTED0.01 s1, 2details
#2--2details
#3ACCEPTED0.01 s1, 2details
#4ACCEPTED0.27 s1, 2details

Code

#include <bits/stdc++.h>

using namespace std;

#define rep(i, a, b) for (int i = a; i < (b); ++i)
#define per(i, a, b) for (int i = a - 1; i >= (b); --i)
#define trav(a, x) for (auto &a : x)

#define all(x) x.begin(), x.end()
#define sz(x) x.size()
#define pb push_back

#define umap unordered_map
#define uset unordered_set
#define INF 1000000000

typedef pair<int, int> ii;
typedef pair<int, ii> iii;
typedef vector<int> vi;
typedef vector<ii> vii;
typedef long long ll;

int main() {
    ios::sync_with_stdio(0);
    cin.tie(NULL);

    pair<string, uset<string>> vis[1001];
    umap<string, int> prev;
    int tc, n;
    cin >> tc;
    while (tc--) {
        cin >> n;
        string nums(n, 0);
        rep(i, 0, n) cin >> nums[i];
        if (prev.count(nums)) {
            cout << vis[prev[nums]].first<<'\n';
            continue;
        }

        queue<string> q;
        q.push(nums);
        vis[tc].second.insert(nums);
        string ans = "NO", goal = string(n, ' ');
        rep(i, 0, n) goal[i] = '1' + i;

        while (!q.empty()) {
            nums = q.front();
            q.pop();
            prev.emplace(nums, tc);
            if (nums == goal) {
                ans = "YES";
                while (!q.empty()) q.pop();
                continue;
            }

            rep(i, 0, n - 3) {
                string next;
                rep(j, i + 2, n - 1) {
                    next = nums;
                    swap(next[i], next[j]);
                    swap(next[i + 1], next[j + 1]);
                    if (!vis[tc].second.count(next)) {
                        vis[tc].second.insert(next);
                        q.push(next);
                    }
                }
            }
        }

        vis[tc].first = ans;
        cout << ans << '\n';
    }
    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
...

Test 2

Group: 2

Verdict:

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

correct output
YES
NO
YES
NO
YES
...

user output
(empty)

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
...

Test 4

Group: 1, 2

Verdict: ACCEPTED

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
NO
NO
YES
NO
YES
...