QOJ.ac

QOJ

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#794023#5466. Permutation CompressionJustJie#WA 1ms3844kbC++203.7kb2024-11-30 09:32:032024-11-30 09:32:04

Judging History

你现在查看的是最新测评结果

  • [2024-11-30 09:32:04]
  • 评测
  • 测评结果:WA
  • 用时:1ms
  • 内存:3844kb
  • [2024-11-30 09:32:03]
  • 提交

answer

/***************************************************************************************************
 * author : Jie Chen (4th Year CS)
 * school : Rochester Institute of Technology
 * created: 11.29.2024 19:49:43
****************************************************************************************************/
#include "bits/stdc++.h"

using namespace std;

using i64 = long long;
using Segment = array<int, 2>;

// Credit: Jiangly
template <typename T>
struct Fenwick {
    int n;
    vector<T> a;
    
    Fenwick(int n = 0) {
        init(n);
    }
    
    void init(int n) {
        this->n = n;
        a.assign(n, T());
    }
    
    // x is 0-index
    void add(int x, T v) {
        for (int i = x + 1; i <= n; i += i & -i) {
            a[i - 1] += v;
        }
    }
    
    // sum from a[0:x]
    T sum(int x) {
        auto ans = T();
        for (int i = x; i > 0; i -= i & -i) {
            ans += a[i - 1];
        }
        return ans;
    }
    
    // sum from a[l:r]
    T rangeSum(int l, int r) {
        return sum(r + 1) - sum(l);
    }
    
    // kth largest sum
    int kth(T k) {
        int x = 0;
        for (int i = 1 << std::__lg(n); i; i /= 2) {
            if (x + i <= n && k >= a[x + i - 1]) {
                x += i;
                k -= a[x - 1];
            }
        }
        return x;
    }
};

void work(int tc) {
    int n, m, k;
    cin >> n >> m >> k;

    vector<int> a(n), pos(n), b(m);

    for (int i = 0; i < n; i++) {
        cin >> a[i];
        a[i]--;
        pos[a[i]] = i;
    }
    for (int i = 0; i < m; i++) {
        cin >> b[i];
        b[i]--;
    }

    multiset<int> t;
    for (int i = 0; i < k; i++) {
        int x;
        cin >> x;
        t.insert(x);
    }

    const auto get = [&](int x) {
        return pos[x];
    };

    int cur = 0;
    vector<bool> del(n);
    for (int i = 0; i < m - 1; i++) {
        int l = get(b[i]);
        int r = get(b[i + 1]);
        if (r < l || l < cur) {
            cout << "NO\n";
            return;
        }
        cur = l;
        for (int j = l + 1; j < r; j++) {
            del[a[j]] = true;
        }
    }

    Fenwick<int> fen(n);
    const auto get2 = [&](int l, int r) {
        return r - l + 1 - fen.rangeSum(l, r);
    };

    set<Segment> s;
    s.insert({0, n - 1});
    for (int tr = n - 1; tr >= 0; tr--) {
        int p = get(tr);
        auto it = prev(s.upper_bound({p, n}));
        auto [l, r] = *it;
        if (del[tr]) {
            if (t.empty()) {
                cout << "NO\n";
                return;
            }
            auto it2 = t.upper_bound(get2(l, r));
            if (it2 == t.begin()) {
                cout << "NO\n";
                return;
            }
            it2--;
            t.erase(it2);
            fen.add(p, 1);
            if (get2(l, r) == 0) {
                s.erase(it);
            }
        } else {
            s.erase(it);
            if (l == r) {
                continue;
            }
            if (p == l) {
                if (get2(l + 1, r)) {
                    s.insert({l + 1, r});
                }
            } else if (p == r) {
                if (get2(l, r - 1)) {
                    s.insert({l, r - 1});
                }
            } else {
                if (get2(l, p - 1)) {
                    s.insert({l, p - 1});
                }
                if (get2(p + 1, r)) {
                    s.insert({p + 1, r});
                }
            }
        }
    }
    cout << "YES\n";
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    int T;
    cin >> T;
    for (int t = 1; T--; t++) {
        work(t);
    }
}

// ~ JustJie

详细

Test #1:

score: 100
Accepted
time: 1ms
memory: 3504kb

input:

3
5 2 3
5 1 3 2 4
5 2
1 2 4
5 5 5
1 2 3 4 5
1 2 3 4 5
1 2 3 4 5
3 2 2
3 1 2
3 2
2 3

output:

YES
YES
NO

result:

ok 3 lines

Test #2:

score: -100
Wrong Answer
time: 0ms
memory: 3844kb

input:

100
2 1 2
2 1
2
1 1
2 1 2
1 2
1
2 2
2 1 1
1 2
1
2
6 1 5
3 4 2 5 6 1
3
5 2 1 1 1
6 1 6
2 1 3 6 4 5
1
4 1 2 2 1 4
3 3 2
2 1 3
2 1 3
2 2
1 1 1
1
1
1
1 1 1
1
1
1
2 1 2
2 1
2
1 2
4 4 3
2 1 3 4
2 1 3 4
4 3 1
1 1 1
1
1
1
6 5 1
6 2 5 4 3 1
6 2 4 3 1
4
1 1 1
1
1
1
6 5 3
3 6 1 4 5 2
3 6 1 4 2
3 3 4
4 3 4
3 4 ...

output:

YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
NO
YES
NO
YES
YES
NO
NO
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
NO
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
YES
Y...

result:

wrong answer 23rd lines differ - expected: 'NO', found: 'YES'