Submission #783207

#	Time	Username	Problem	Language	Result	Execution time	Memory
783207		teokakabadze	Connecting Supertrees (IOI20_supertrees)	C++17	0 / 100	154 ms	26964 KiB

supertrees

#include "supertrees.h"
#include<bits/stdc++.h>
#define pb push_back
using namespace std;

int a[1005][1005], i, j, f[1005], k, c[1005], d, l, cnt, pr[1005];
vector<int> s;

int construct(std::vector<std::vector<int>> p)
{
    int b;
	int n = p.size();
	std::vector<std::vector<int>> answer;
	for(i = 0; i < n; i++)
    for(j = 0; j < n; j++)
    {
        if(p[i][j] == 3) return 0;
        if(p[i][j] == 2) d = 1;
    }
	for(i = 0; i < n; i++)
    if(!f[i])
    {
        if(!c[i]) c[i] = ++k;
        for(j = 0; j < n; j++)
        if(!f[j] && p[i][j] == 1) c[j] = c[i], f[j] = 1;
    }
    for(i = 0; i < n; i++)
    for(j = 0; j < n; j++)
    {
        if(c[i] == c[j] && p[i][j] != 1) return 0;
        if(c[i] == c[j] && i != j) a[i][j] = a[j][i] = 1;
    }
    for(i = 0; i < n; i++)
    f[i] = 0;
    if(d)
    {
        for(i = 0; i < n; i++)
        if(!f[i])
        {
            f[i] = 1;
            vector<int> e, r;
            for(j = 0; j < n; j++)
            if(!f[j] && a[i][j]) { f[j] = 1, pr[j] = i; e.pb(j); }
            l = i, cnt = 0;
            r.pb(i);
            for(j = 0; j < n; j++)
            if(!f[j] && p[i][j] == 2)
            {
                r.pb(j);
                f[j] = 1, a[l][j] = a[j][l] = 1, l = j;
                for(b = 0; b < n; b++)
                if(!f[b] && a[j][b]) { f[b] = 1, pr[b] = j; e.pb(b); }
                cnt++;
            }
            if(cnt < 2) return 0;
            a[i][l] = a[l][i] = 1;
            for(auto x : r)
            for(auto y : r)
            if(x != y && p[x][y] != 2) return 0;
            for(auto x : r)
            for(auto y : e)
            {
                if(x != y && x != pr[y] && p[x][y] != 2) return 0;
                if(x != y && x == pr[y] && p[x][y] != 1) return 0;
            }
            for(auto x : e)
            for(auto y : e)
            {
                if(x != y && p[x] != p[y] && p[x][y] != 2) return 0;
                if(x != y && p[x] == p[y] && p[x][y] != 1) return 0;
            }
        }
    }
	for(i = 0; i < n; i++)
    {
        for(j = 0; j < n; j++)
        s.push_back(a[i][j]);
        answer.push_back(s);
        s.clear();
    }
	build(answer);
	return 1;
}

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