Submission #403791

#	Time	Username	Problem	Language	Result	Execution time	Memory
403791		victoriad	Connecting Supertrees (IOI20_supertrees)	C++14	100 / 100	280 ms	26552 KiB

supertrees

#include "supertrees.h"
#include <vector>
#include <cmath>
#include <cstdio>
using namespace std;


int construct(std::vector<std::vector<int> > p) {
	int n = p.size();
	vector<vector<int> >r(n);
	vector<vector<int> >c;
    vector<int>us(n,-1);
    int cc=0;
    vector<vector<int> >co;
    vector<int>u(n,-1);
    int aa=0;
    for(int i=0;i<n;i++){
        if(u[i]==-1){
            vector<int>s=p[i];
            co.push_back(s);
            for(int k=0;k<n;k++){
                if(s[k]!=0){
                    if(u[k]==-1){
                        u[k]=aa;
                    }
                    else if(u[k]!=aa){
                        return 0;
                        break;
                    }
                }
            }
            aa++;
        }
        else{
            int x=u[i];
            for(int k=0;k<n;k++){
                if(co[x][k]==p[i][k])continue;
                else{
                    if(co[x][k]==0 && p[i][k]!=0){
                        return 0;
                        break;
                    }
                    if(co[x][k]!=0 && p[i][k]==0){
                        return 0;
                        break;
                    }
                }
            }
        }
    }
    for(int i=0;i<n;i++){
        for(int k=0;k<n;k++){
            if(p[k][i]!=p[i][k]||p[k][i]==3){
                return 0;
                break;
            }
        }
    }
    for(int i=0;i<n;i++){
        if(p[i][i]!=1){
            return 0;
            break;
        }
        if(us[i]==-1){
            vector<int>s=p[i];
            vector<int>u;
            for(int k=0;k<n;k++){
                if(s[k]!=0){
                    if(us[k]==-1){
                    us[k]=cc;
                    u.push_back(k);
                    }
                    else{
                        return 0;
                        break;
                    }
                }
            }
            c.push_back(u);
            cc++;
        }
    }
    vector<int>a(n,0);
    for(int i=0;i<n;i++)r[i]=a;
    int pa=0;
    for(int i=0;i<n;i++){
        for(int k=0;k<n;k++){
            if(p[i][k]!=0 && us[i]!=us[k]){
                return 0;
                break;
            }
        }
    }
    for(int i=0;i<c.size();i++){
        vector<int>padre(n,-1);
        vector<int>raiz;
        for(int k=0;k<c[i].size();k++){
            if(padre[c[i][k]]==-1){
                padre[c[i][k]]=c[i][k];
                pa=c[i][k];
                int x=pa;
                raiz.push_back(pa);
                vector<int>s=p[c[i][k]];
                for(int j=0;j<n;j++){
            
                    if(p[c[i][k]][j]==1){
                        if(p[j]!=s){
                            return 0;
                            break;
                        }
                        if(j!=c[i][k] && padre[j]!=-1){
                            return 0;
                            break;
                        }
                        padre[j]=pa;
                        r[x][j]=1;
                        r[j][x]=1;
                        x=j;
                    }
                }
            }
        }
        int x=-1;
        if(raiz.size()==2){
            return 0;
        }
        for(int k=0;k<raiz.size();k++){
            if(x==-1){
                x=raiz[k];
            }
            else{
                r[x][raiz[k]]=1;
                r[raiz[k]][x]=1;
                x=raiz[k];
            }
        }
        if(x!=raiz[0]){
            r[raiz[0]][x]=1;
            r[x][raiz[0]]=1;
        }


    }
    for(int i=0;i<n;i++)r[i][i]=0;
    
    build(r);
    return 1;
}

Compilation message (stderr)

supertrees.cpp: In function 'int construct(std::vector<std::vector<int> >)':
supertrees.cpp:94:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::vector<int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   94 |     for(int i=0;i<c.size();i++){
      |                 ~^~~~~~~~~
supertrees.cpp:97:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   97 |         for(int k=0;k<c[i].size();k++){
      |                     ~^~~~~~~~~~~~
supertrees.cpp:127:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  127 |         for(int k=0;k<raiz.size();k++){
      |                     ~^~~~~~~~~~~~

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