Submission #398556

#	Time	Username	Problem	Language	Result	Execution time	Memory
398556		priority_queue	Connecting Supertrees (IOI20_supertrees)	C++14	100 / 100	272 ms	28100 KiB

supertrees

#include <bits/stdc++.h>

using namespace std;

#define forint(i, N) for (int i = 0; i < (N); i++)

using namespace std;

void build(std::vector<std::vector<int> > b);

int construct(vector<std::vector<int> > p) {

	int N = p.size();

	vector< vector<int> > b(N, vector<int>(N, 0));

	vector<bool> visited(N, false);

	vector<std::vector<int> > p2(N, vector<int>(N, 0));

	forint(i, N) {

        if (!visited[i]) { // create new. All connected to i also not visited
            
            visited[i] = true;
        	
        	vector< vector<int> > circle(1);
			circle[0].push_back(i);
	
        	vector<int> path2 = {};
            
        	for (int j = i+1; j < N; j++) {
        		
        		if (p[i][j] == 1) {
        			if (visited[j]) {
        				return 0;
        			}
        			b[i][j] = 1;
        			b[j][i] = 1;
        			visited[j] = true;

        			circle[0].push_back(j);
//cerr << "he -->" << circle[0].size() << endl;
        			continue;
        		}

        		if (p[i][j] == 2) {
        			if (visited[j]) {
        				return 0;
        			}
        			path2.push_back(j);
        			continue;
        		}

        		if (p[i][j] == 3) {
        			return 0;
        		}
        	}

        	if (!path2.empty()) {
        		if (path2.size() < 2) {
        			return 0;
        		}

        		int last = i;

        		forint(j, path2.size()) {

        			if (!visited[path2[j]]) {
                		// make it root
        				b[last][path2[j]] = 1;
        				b[path2[j]][last] = 1;
        				last = path2[j];
        				visited[path2[j]] = true;

        				circle.push_back({path2[j]});

        				for (int j2 = j+1; j2 < path2.size(); j2++) {
        					if (!visited[path2[j2]] && p[path2[j]][path2[j2]] == 1) {
        						b[path2[j]][path2[j2]] = 1;
        						b[path2[j2]][path2[j]] = 1;
        						visited[path2[j2]] = true;

        						circle[circle.size()-1].push_back(path2[j2]);
        					}	
        				}
        			}
        		}
        		b[i][last] = 1;
        		b[last][i] = 1;
        	}
/*
        	for (auto& c : circle) {
        		for (auto d : c) {
        			cout << d << " ";
        		}
        		cout << endl;
        	}
*/
        	forint(c, circle.size()) {
        		for (int d1 = 0; d1 < circle[c].size(); d1++) {
        			for (int d2 = 0; d2 < circle[c].size(); d2++) {
        				p2[circle[c][d1]][circle[c][d2]] = 1;
        			}	
        		}

        		forint(d, circle.size()) {
        			if (c!=d) {
		        		for (int d1 = 0; d1 < circle[c].size(); d1++) {
        					for (int d2 = 0; d2 < circle[d].size(); d2++) {
        						p2[circle[c][d1]][circle[d][d2]] = 2;
	        				}	
    		    		}
        			}
        		}
        	}
        }
    }

    forint(i, N) {
    	forint(j, N) {
    		if (p2[i][j] != p[i][j]) {
    			return 0;
    		}
    	}
    }
    
    build(b);

    return 1;

}

Compilation message (stderr)

supertrees.cpp: In function 'int construct(std::vector<std::vector<int> >)':
supertrees.cpp:5:40: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
    5 | #define forint(i, N) for (int i = 0; i < (N); i++)
      |                                        ^
supertrees.cpp:67:11: note: in expansion of macro 'forint'
   67 |           forint(j, path2.size()) {
      |           ^~~~~~
supertrees.cpp:78:35: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   78 |             for (int j2 = j+1; j2 < path2.size(); j2++) {
      |                                ~~~^~~~~~~~~~~~~~
supertrees.cpp:5:40: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::vector<int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
    5 | #define forint(i, N) for (int i = 0; i < (N); i++)
      |                                        ^
supertrees.cpp:100:10: note: in expansion of macro 'forint'
  100 |          forint(c, circle.size()) {
      |          ^~~~~~
supertrees.cpp:101:31: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  101 |           for (int d1 = 0; d1 < circle[c].size(); d1++) {
      |                            ~~~^~~~~~~~~~~~~~~~~~
supertrees.cpp:102:32: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  102 |            for (int d2 = 0; d2 < circle[c].size(); d2++) {
      |                             ~~~^~~~~~~~~~~~~~~~~~
supertrees.cpp:5:40: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::vector<int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
    5 | #define forint(i, N) for (int i = 0; i < (N); i++)
      |                                        ^
supertrees.cpp:107:11: note: in expansion of macro 'forint'
  107 |           forint(d, circle.size()) {
      |           ^~~~~~
supertrees.cpp:109:33: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  109 |             for (int d1 = 0; d1 < circle[c].size(); d1++) {
      |                              ~~~^~~~~~~~~~~~~~~~~~
supertrees.cpp:110:34: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  110 |              for (int d2 = 0; d2 < circle[d].size(); d2++) {
      |                               ~~~^~~~~~~~~~~~~~~~~~

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