Submission #487864

#	Time	Username	Problem	Language	Result	Execution time	Memory
487864		ala2	Connecting Supertrees (IOI20_supertrees)	C++14	40 / 100	239 ms	26144 KiB

supertrees

#include "supertrees.h"
#include <vector>
#include<iostream>
using namespace std;
int pa[1001000];
int sz[1001000];
int root(int x)
{
    while(x!=pa[x])
    {
        pa[x]=pa[pa[x]];
        x=pa[x];
    }
    return x;
}

void unit(int x,int y)
{
    x=root(x);
    y=root(y);
    if(x==y)
        return ;
    if(sz[x]>sz[y])
    {
        sz[x]+=sz[y];
        pa[y]=x;
    }
    else
    {
        sz[y]+=sz[x];
        pa[x]=y;
    }
}

vector<int>v[1010];
int vi[1010];
int vis[1010][1010];
int construct(vector<vector<int>> p) {
	int n = p.size();
	int bb=0;
	vector<vector<int>> answer;
	for (int i = 0; i < n; i++) {
		vector<int> row;
		row.resize(n);
		answer.push_back(row);
	}
	for(int i=0;i<n;i++)
    {
        pa[i]=i;
        sz[i]=1;
    }
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<n;j++)
        {
            if(p[i][j]==1)
            {
                unit(i,j);
            }
        }
    }
    for(int i=0;i<n;i++)
    {
        v[root(i)].push_back(i);
    }
    for(int i=0;i<n;i++)
    {
        if(i!=root(i))
            continue;
        if(v[i].size()==1)
        {
            continue;
        }
        for(int j=0;j<v[i].size()-1;j++)
        {
            int x=v[i][j];
            int y=v[i][j+1];
            answer[x][y]=1;
            answer[y][x]=1;
        }
    }
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<n;j++)
        {
            if(p[i][j]!=1&&root(i)==root(j))
                bb=1;
            if(p[i][j]==1&&root(i)!=root(j))
                bb=1;

        }
    }
    for(int i=0;i<n;i++)
    {
        vector<int>r;
        r.push_back(i);
        vi[root(i)]=1;
        for(int j=0;j<n;j++)
        {
            if(p[i][j]==2&&!vi[root(j)])
            {
                r.push_back(j);
                vi[root(j)]=1;
            }
        }
       // cout<<"           "<<r.size()<<endl;
        if(r.size()==2)
        bb=1;
        if(r.size()==1)
            continue;
        for(int k=0;k<r.size()-1;k++)
        {
            int x=r[k];
            int y=r[k+1];
            answer[x][y]=1;
            answer[y][x]=1;
        }
        int ss=r.size()-1;
        answer[r[ss]][r[0]]=1;
        answer[r[0]][r[ss]]=1;
        for(int ii=0;ii<r.size();ii++)
        {
            for(int jj=0;jj<r.size();jj++)
            {
                if(ii==jj)
                    continue;
                vis[r[ii]][r[jj]]=1;
                vis[r[jj]][r[ii]]=1;
            }
        }

    }
    for(int i=0;i<n;i++)
    {
        answer[i][i]=0;
        for(int j=0;j<n;j++)
        {
         //   cout<<vis[i][j]<<" ";
            if(p[i][j]==2&&!vis[i][j])
                bb=1;
            if(p[i][j]!=2&&vis[i][j])
                bb=1;
            if(p[i][j]==1&&root(i)!=root(j))
                bb=1;
            if(root(i)==root(j)&&p[i][j]!=1)
                bb=1;
            if(p[i][j]==0&&vis[i][j])
                bb=1;
        }
     
    }

    if(bb)
        return 0;
    else
    {
	build(answer);
	return 1;
    }

}

Compilation message (stderr)

supertrees.cpp: In function 'int construct(std::vector<std::vector<int> >)':
supertrees.cpp:74:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   74 |         for(int j=0;j<v[i].size()-1;j++)
      |                     ~^~~~~~~~~~~~~~
supertrees.cpp:111:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  111 |         for(int k=0;k<r.size()-1;k++)
      |                     ~^~~~~~~~~~~
supertrees.cpp:121:24: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  121 |         for(int ii=0;ii<r.size();ii++)
      |                      ~~^~~~~~~~~
supertrees.cpp:123:28: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  123 |             for(int jj=0;jj<r.size();jj++)
      |                          ~~^~~~~~~~~

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