Submission #399444

#	Time	Username	Problem	Language	Result	Execution time	Memory
399444		bipartite_matching	Connecting Supertrees (IOI20_supertrees)	C++14	100 / 100	259 ms	28180 KiB

supertrees

#include <bits/stdc++.h>

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

using namespace std; 

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

int construct(vector< vector<int> > p) {
	int n = p.size();

	//vector<int> coor(n, -1);
	

	vector< vector<int> > ans(n, vector<int>(n, 0));

	vector<int> visited(n, false);

	vector< vector<int> > p2(n, vector<int>(n, 0));

	forint(i, n) {
		//cerr << i << endl;
		if (!visited[i]) {
			vector< vector<int> > tree;

			visited[i] = true;
		//if (coor[i] == -1) {
			//coor[i] == i;
			tree.push_back({i});

			vector<int> tmp;
			//tmp.push_back(i);

			for (int j = i + 1; j < n; j++) {
				if (p[i][j] == 3) {
					return 0;
				}
				if (p[i][j] == 1) {
					
					//if (coor[j] != -1) {
					if (visited[j]) {
						return 0;
					}

					//coor[j] = i;
					ans[i][j] = 1;
					ans[j][i] = 1;

					visited[j] = true;
					tree[tree.size()-1].push_back(j);

				}
				if (p[i][j] == 2) {
					//cerr << "eh " << j << endl;
					tmp.push_back(j);
				}

			}

			if (tmp.size() == 1) {
				return 0;
			}

			if (tmp.size() > 1) {
				int last_node = i;
				for (int k = 0; k < tmp.size(); k++) {
					if (!visited[tmp[k]]) {

						visited[tmp[k]] =  true;

						tree.push_back({tmp[k]});

						ans[last_node][tmp[k]] = 1;
						ans[tmp[k]][last_node] = 1;

						last_node = tmp[k];

						for (int k2 = k+1; k2 < tmp.size(); k2++) {
							if (!visited[tmp[k2]] && p[tmp[k]][tmp[k2]] == 1) {
								visited[tmp[k2]] = true;
								ans[tmp[k]][tmp[k2]] = 1;
								ans[tmp[k2]][tmp[k]] = 1;
								tree[tree.size()-1].push_back(tmp[k2]);
							}
						}
					}
				}

				ans[last_node][i] = 1;
				ans[i][last_node] = 1;
			}

			//tree.push_back(tmp);
			/*
			cerr << i << "-----" << endl;
			for (auto a : tree) {
				for (auto b: a) {
					cerr << b << " ";
				}
				cerr << endl;
			}
			cerr << "-----" << endl;
			*/

			for (int k = 0; k < tree.size(); k++) {

				for (auto a : tree[k]) {
					for (auto b : tree[k]) {
						p2[a][b] = 1;
					}
				}

				for (int k2 = k+1; k2 < tree.size(); k2++) {
					for (auto a : tree[k]) {
						for (auto b : tree[k2]) {
							p2[a][b] = 2;
							p2[b][a] = 2;
						}
					}
				}
			}
		}
		/*
		else {
			forint(j, n) {
				if (p[i][j] != p[coor[i]][j]) {
					return 0;
				}
			}
		}
		*/

	}


	forint(i, n) {
		forint(j, n) {			
			if (p[i][j] != p2[i][j]) {
				
//cerr << i << " " << j << "---" << p2[i][j] << endl;

				return 0;

			}

		}
	}

	build(ans);

	return 1;

}

Compilation message (stderr)

supertrees.cpp: In function 'int construct(std::vector<std::vector<int> >)':
supertrees.cpp:66:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   66 |     for (int k = 0; k < tmp.size(); k++) {
      |                     ~~^~~~~~~~~~~~
supertrees.cpp:78:29: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   78 |       for (int k2 = k+1; k2 < tmp.size(); k2++) {
      |                          ~~~^~~~~~~~~~~~
supertrees.cpp:105:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::vector<int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  105 |    for (int k = 0; k < tree.size(); k++) {
      |                    ~~^~~~~~~~~~~~~
supertrees.cpp:113:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::vector<int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  113 |     for (int k2 = k+1; k2 < tree.size(); k2++) {
      |                        ~~~^~~~~~~~~~~~~

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