제출 #1073022

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1073022		mc061	슈퍼트리 잇기 (IOI20_supertrees)	C++14	46 / 100	153 ms	24328 KiB

supertrees

#include <bits/stdc++.h>
using namespace std;

const int N = 1001;

struct DSU {
    vector<int> p;
    vector<vector<int>> els;
    DSU() {
        p.resize(N);
        els.resize(N);
        iota(p.begin(), p.end(), 0);
        for (int i = 0; i < N; ++i) {
            els[i].push_back(i);
        }
    }
    int get(int a) {
        return p[a] = (a == p[a] ? a : get(p[a]));
    }
    void merge(int a, int b) {
        int x = get(a);
        int y = get(b);
        if (x == y) return;
        if (els[x].size() > els[y].size()) {
            swap(x, y);
        }
        p[x] = y;
        while (els[x].size()) els[y].push_back(els[x].back()), els[x].pop_back();
    }
};

void build(vector<vector<int>> b);
int construct(vector<vector<int>> p) {
    DSU lines;
    const int n = p.size();
    vector<vector<int>> b(n, vector<int>(n, 0));
    for (int i = 0; i < n; ++i) {
        for (int j = i+1; j < n; ++j) {
            if (p[i][j] == 1) {
                int x = lines.get(i);
                int y = lines.get(j);
                if (x != y) {
                    b[i][j] = b[j][i] = 1;
                    lines.merge(x, y);
                }
            }
        }
    }
    DSU comps;
    for (int i = 0; i < n; ++i) {
        for (int j = i+1; j < n; ++j) {
            if (p[i][j] > 0) comps.merge(i, j);
        }
    }
    vector<bool> vis(n, false);
    for (int i = 0; i < n; ++i) {
        int j = comps.get(i);
        if (vis[j]) continue;
        vis[j] = true;
        vector<int> vs = comps.els[j];
        vector<int> twos;
        vector<bool> vis_lines(n, false);
        for (int k = 0; k < vs.size(); ++k) {
            int v = vs[k];
            if (vis_lines[lines.get(v)]) continue;
            int u = lines.els[lines.get(v)].front();
            twos.push_back(u);
            vis_lines[lines.get(v)] = true;
        }
        if (twos.size() == 1) continue;
        for (int k = 0; k < twos.size(); ++k) {
            int nxt = (k+1)%twos.size();
            b[twos[k]][twos[nxt]] = b[twos[nxt]][twos[k]] = 1;
        }
    }
    build(b);
    return true;
}

컴파일 시 표준 에러 (stderr) 메시지

supertrees.cpp: In function 'int construct(std::vector<std::vector<int> >)':
supertrees.cpp:63:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   63 |         for (int k = 0; k < vs.size(); ++k) {
      |                         ~~^~~~~~~~~~~
supertrees.cpp:71:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   71 |         for (int k = 0; k < twos.size(); ++k) {
      |                         ~~^~~~~~~~~~~~~

• Subtask 1: 11 / 11
• Subtask 2: 0 / 10
• Subtask 3: 0 / 19
• Subtask 4: 35 / 35
• Subtask 5: 0 / 21
• Subtask 6: 0 / 4