Submission #1004693

#	Time	Username	Problem	Language	Result	Execution time	Memory
1004693		spensa	슈퍼트리 잇기 (IOI20_supertrees)	C++17	0 / 100	1 ms	428 KiB

supertrees

#include "supertrees.h"
#include <iostream>
#include <vector>
#include <map>
#include <set>
using namespace std;

int construct(std::vector<std::vector<int>> p) {
    //subtask 3: mini cycles, but as connected components
	int n = p.size();
    vector<vector<int>> ans(n, vector<int> (n));  

    for(int i=0; i<n; i++){
        for(int j=0; j<n; j++){
            ans[i][j] = 0;
        }
    }

    map<int, vector<int>> cc;
    vector<int> flags;
    flags.push_back(0);
    cc[0].push_back(0);
    for(int i=1; i<n; i++){
        int tmp = 0;
        for(int j: flags){
            if(p[i][j]==0) continue;
            if(i==j) continue;
            ans[i][j] = 1;
            ans[j][i] = 1;
            cc[j].push_back(i);
            tmp++;
            // break;
        }
        if(tmp>=2) return 0;
        if(tmp==0){
            flags.push_back(i);
            cc[i].push_back(i);
        }
    }



    set<pair<int,int>> act;

    for(auto i: cc){
        // for(int j: i.second) cout<<j<<" ";
        // cout<<"\n";
        int sz = i.second.size();
        if(sz<3) continue;
        ans[i.second[sz-1]][i.second[sz-2]] = 1;
        ans[i.second[sz-2]][i.second[sz-1]] = 1;
        act.insert(make_pair(i.second[sz-2], i.second[sz-1]));
        act.insert(make_pair(i.second[sz-1], i.second[sz-2]));
    
    //     cout<<i.second[-1]<<"!!"<<i.second[-2]<<"\n";
    //     ans[i.second[-1]][i.second[-2]] = 1;
    //     ans[i.second[-2]][i.second[-1]] = 1;
    }

    for(auto i: cc){
        // cout<<"i.first="<<i.first<<"\n";
        for(int j: i.second){
            for(int k: i.second){
                act.insert(make_pair(j, k));
                act.insert(make_pair(k, j));
            }
        }
    }

    // for(auto i: act) cout<<i.first<<" "<<i.second<<"\n";
    // cout<<"\n";

    // sort(act.begin(), act.end());
    for(int i=0; i<n; i++){
        for(int j=0; j<n; j++){
            // cout<<"here";
            if(i==j) continue;
            if(p[i][j]==0){
                if(act.count({i,j})==1) return 0;
            }
            else{
                if(act.count({i,j})==0) return 0;
            }
        }
    }

    build(ans);
	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