Submission #728127

#	Time	Username	Problem	Language	Result	Execution time	Memory
728127		hoainiem	Connecting Supertrees (IOI20_supertrees)	C++14	21 / 100	215 ms	23884 KiB

supertrees

#include "supertrees.h"
#include <bits/stdc++.h>
using namespace std;
int n, flag[1008];
bool kt[1008];
vector<int>ds, cir, tour, vt[1008];
vector<vector<int> >a;
int construct(std::vector<std::vector<int>> p) {
    n = p.size();
    a.resize(n);
    for (int i = 0; i < n; i++){
        for (int j = i + 1; j < n; j++)
            assert(p[i][j] < 3);
        a[i].resize(n, 0);
    }
    fill(kt, kt + n, false);
    for (int i = 0; i < n; i++)
        if (!kt[i]){
            ds.push_back(i);
            vt[i].push_back(i);
            kt[i] = true;
            flag[i] = i;
            for (int j = i + 1; j < n; j++)
                if (!kt[j] && p[i][j] == 1){
                    vt[i].push_back(j);
                    kt[j] = true;
                    flag[j] = i;
                }
            for (int u : vt[i])
                for (int v : vt[i])
                    if (p[u][v] != 1)
                        return 0;
            for (int pos = 1; pos < (int)vt[i].size(); pos++)
                a[vt[i][pos]][vt[i][pos - 1]] = a[vt[i][pos - 1]][vt[i][pos]] = 1;

        }
    for (int i = 0; i < n; i++)
        for (int j = i + 1; j < n; j++)
            if (p[i][j] == 1 && flag[i] != flag[j])
                return 0;
    fill(kt, kt + n, false);
    for (int i : ds)
        if (!kt[i]){
            cir.clear();
            tour.clear();
            kt[i] = true;
            tour.push_back(i);
            for (int tmp : vt[i])
                cir.push_back(tmp);
            for (int j : ds)
                if (p[i][j] == 2){
                    if (kt[j])
                        return 0;
                    tour.push_back(j);
                    kt[j] = true;
                    for (int u : vt[j])
                        for (int v : cir)
                            if (p[u][v] != 2)
                                return 0;
                    for (int u : vt[j])
                        cir.push_back(u);
                }
            if (tour.size() <= 2){
                if (tour.size() == 2)
                    return 0;
                else
                    break;
            }
            tour.push_back(tour[0]);
            for (int pos = 1; pos < (int)tour.size(); pos++)
                a[tour[pos]][tour[pos - 1]] = a[tour[pos - 1]][tour[pos]] = 1;
        }
    build(a);
    return 1;
}

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