QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#369441#6503. DFS Order 3bigJWA 140ms3576kbC++202.4kb2024-03-28 09:46:562024-03-28 09:46:58

Judging History

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

  • [2024-03-28 09:46:58]
  • 评测
  • 测评结果:WA
  • 用时:140ms
  • 内存:3576kb
  • [2024-03-28 09:46:56]
  • 提交

answer

#include <bits/stdc++.h>

using i64 = int64_t;

namespace stdr = std::ranges;
namespace stdv = std::views;
struct DSU {
    std::vector<int> f, siz;

    DSU() {}
    DSU(int n) {
        init(n);
    }

    void init(int n) {
        f.resize(n);
        std::iota(f.begin(), f.end(), 0);
        siz.assign(n, 1);
    }

    constexpr int find(int x) {
        while (x != f[x]) {
            x = f[x] = f[f[x]];
        }
        return x;
    }
    constexpr int operator[](int i) {
        return find(i);
    }

    bool same(int x, int y) {
        return find(x) == find(y);
    }

    bool merge(int x, int y, bool t = true) {
        x = find(x);
        y = find(y);
        if (x == y) {
            return false;
        }
        if (t && siz[x] < siz[y]) {
            std::swap(x, y);
        }
        siz[x] += siz[y];
        f[y] = x;
        return true;
    }

    int size(int x) {
        return siz[find(x)];
    }
};
void solve() {
    int n;
    std::cin >> n;

    std::vector a(n, std::vector<int>(n));
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < n; j++) {
            std::cin >> a[i][j];
            a[i][j]--;
        }
    }

    if (n == 1) {
        return;
    }

    std::set<std::pair<int, int>> S;
    DSU dsu(n);

    for (auto b : a) {
        if (b[0] > b[1]) {
            std::swap(b[0], b[1]);
        }
        S.emplace(b[0], b[1]);
        dsu.merge(b[0], b[1]);
    }

    std::set<int> has;
    for (auto b : a) {
        for (int i = 0; i + 1 < n; i++) {
            if (!dsu.same(b[i], b[i + 1])) {
                has.insert(b[i + 1]);
            }
        }
    }

    for (int i = 0; i < n; i++) {
        int u = -1;
        for (int j = 0; j + 1 < n; j++) {
            if (has.contains(a[i][j])) {
                u = a[i][j];
            }
            if (!dsu.same(a[i][j], a[i][j + 1])) {
                dsu.merge(a[i][j], a[i][j + 1]);
                auto v = a[i][j + 1];
                if (u > v) {
                    std::swap(u, v);
                }
                S.emplace(u, v);
                break;
            }
        }
    }

    for (auto [x, y] : S) {
        std::cout << x + 1 << " " << y + 1 << "\n";
    }
}

int main() {
    std::cin.tie(nullptr);
    std::ios::sync_with_stdio(false);

    int t;
    std::cin >> t;

    while (t--) {
        solve();
    }

    return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 0ms
memory: 3548kb

input:

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

output:

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

result:

ok correct answer! (4 test cases)

Test #2:

score: -100
Wrong Answer
time: 140ms
memory: 3576kb

input:

20000
10
1 2 4 5 6 7 3 8 10 9
2 1 4 5 6 7 3 8 10 9
3 8 1 2 4 5 6 7 10 9
4 5 6 7 1 2 3 8 10 9
5 4 6 7 1 2 3 8 10 9
6 7 4 5 1 2 3 8 10 9
7 6 4 5 1 2 3 8 10 9
8 3 1 2 4 5 6 7 10 9
9 10 1 2 4 5 6 7 3 8
10 1 2 4 5 6 7 3 8 9
10
1 4 3 8 2 9 6 5 7 10
2 8 9 6 3 4 1 5 7 10
3 8 2 9 6 4 1 5 7 10
4 1 3 8 2 9 6 5...

output:

1 2
1 3
1 4
1 10
3 8
4 5
4 6
6 7
9 10
1 4
2 8
3 4
3 8
4 5
5 7
6 9
7 10
8 9
1 9
2 4
2 8
3 10
5 6
5 7
5 9
8 9
8 10
1 6
2 4
2 10
3 8
5 7
6 7
6 9
6 10
7 8
1 5
1 9
2 10
3 6
3 7
3 10
4 7
5 7
8 9
1 10
2 6
3 8
4 8
5 10
6 10
7 9
8 9
8 10
1 10
2 3
2 9
2 10
3 7
4 8
5 7
5 8
6 7
1 4
1 9
2 3
2 4
4 7
4 8
5 10
6 7
...

result:

wrong answer ord[1] didn't pass the test (test case 3)