oj.uz

Submission #834706

# Submission time Handle Problem Language Result Execution time Memory
834706 2023-08-22T17:29:46 Z Josia Simurgh (IOI17_simurgh) C++17
30 / 100
 239 ms 3796 KB 
#include "simurgh.h"
#include <bits/stdc++.h>

using namespace std;





int query(set<int> x) {
	vector<int> edges;
	for (int i: x) edges.push_back(i);
	return count_common_roads(edges);
}



int n, m;

vector<vector<int>> graphList;
vector<vector<int>> graphMatrix;
vector<vector<int>> knownEdges;


struct solve {
	vector<int> vis;
	set<int> usedEdges;

	void dfs(int v, int p, int forbidden, int col) {
		if (vis[v]!=-1) return;
		vis[v] = col;

		if (p!=-1) {
			usedEdges.insert(graphMatrix[v][p]);
		}

		for (int u: graphList[v]) {
			if (u == forbidden) continue;
			dfs(u, v, forbidden, col);
		}
	}


	void start(int v) {
		vis.assign(n, -1);
		usedEdges.clear();
		int col = 0;
		vector<vector<int>> colors;
		for (int u: graphList[v]) {
			if (vis[u]==-1) {
				dfs(u, -1, v, col);
				col++;
				colors.push_back({});
			}

			colors[vis[u]].push_back(u);
		}

		for (auto i: colors) usedEdges.insert(graphMatrix[v][i[0]]);

		for (auto i: colors) {
			vector<int> vals;
			usedEdges.erase(graphMatrix[v][i[0]]);
			for (int j: i) {
				// if (knownEdges[v][j]) continue;
				usedEdges.insert(graphMatrix[v][j]);
				vals.push_back(query(usedEdges));
				usedEdges.erase(graphMatrix[v][j]);
			}
			usedEdges.insert(graphMatrix[v][i[0]]);
			int hi = *max_element(vals.begin(), vals.end());

			for (int j = 0; j<(int)i.size(); j++) {
				if (vals[j] == hi) {
					knownEdges[i[j]][v] = 1;
					knownEdges[v][i[j]] = 1;
				}
				else {
					knownEdges[i[j]][v] = 0;
					knownEdges[v][i[j]] = 0;
				}
			}
		}
	}


	vector<int> output;
	solve() {
		for (int i = 0; i<n; i++) {
			start(i);
		}
		output.clear();

		for (int i = 0; i<n; i++) {
			for (int j = i+1; j<n; j++) {
				if (knownEdges[i][j]==0) continue;
				if (knownEdges[i][j]==-1) continue;
				// assert(knownEdges[i][j]==1);

				output.push_back(graphMatrix[i][j]);
			}
		}
	}
};



vector<int> find_roads(int _n, vector<int> u, vector<int> v) {
	n = _n;
	m = u.size();
	graphList.assign(n, vector<int>());
	graphMatrix.assign(n, vector<int>(n, -1));
	knownEdges.assign(n, vector<int>(n, -1));

	for (int i = 0; i<m; i++) {
		graphList[u[i]].push_back(v[i]);
		graphList[v[i]].push_back(u[i]);

		graphMatrix[u[i]][v[i]] = i;
		graphMatrix[v[i]][u[i]] = i;
	}

	solve sol;

	return sol.output;
}

Subtask #1
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 212 KB correct
2 Correct 0 ms 212 KB correct
3 Correct 1 ms 336 KB correct
4 Correct 0 ms 212 KB correct
5 Correct 0 ms 212 KB correct
6 Correct 1 ms 212 KB correct
7 Correct 1 ms 212 KB correct
8 Correct 0 ms 212 KB correct
9 Correct 0 ms 212 KB correct
10 Correct 0 ms 212 KB correct
11 Correct 1 ms 212 KB correct
12 Correct 0 ms 212 KB correct
13 Correct 1 ms 212 KB correct

Subtask #2
17.0 / 17.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 212 KB correct
2 Correct 0 ms 212 KB correct
3 Correct 1 ms 336 KB correct
4 Correct 0 ms 212 KB correct
5 Correct 0 ms 212 KB correct
6 Correct 1 ms 212 KB correct
7 Correct 1 ms 212 KB correct
8 Correct 0 ms 212 KB correct
9 Correct 0 ms 212 KB correct
10 Correct 0 ms 212 KB correct
11 Correct 1 ms 212 KB correct
12 Correct 0 ms 212 KB correct
13 Correct 1 ms 212 KB correct
14 Correct 6 ms 340 KB correct
15 Correct 5 ms 340 KB correct
16 Correct 5 ms 340 KB correct
17 Correct 5 ms 340 KB correct
18 Correct 2 ms 340 KB correct
19 Correct 5 ms 340 KB correct
20 Correct 5 ms 340 KB correct
21 Correct 4 ms 340 KB correct
22 Correct 4 ms 340 KB correct
23 Correct 3 ms 340 KB correct
24 Correct 3 ms 340 KB correct
25 Correct 1 ms 212 KB correct
26 Correct 3 ms 340 KB correct
27 Correct 3 ms 340 KB correct
28 Correct 2 ms 340 KB correct
29 Correct 2 ms 212 KB correct
30 Correct 3 ms 340 KB correct
31 Correct 3 ms 340 KB correct
32 Correct 3 ms 340 KB correct
33 Correct 3 ms 340 KB correct

Subtask #3
0 / 21.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 212 KB correct
2 Correct 0 ms 212 KB correct
3 Correct 1 ms 336 KB correct
4 Correct 0 ms 212 KB correct
5 Correct 0 ms 212 KB correct
6 Correct 1 ms 212 KB correct
7 Correct 1 ms 212 KB correct
8 Correct 0 ms 212 KB correct
9 Correct 0 ms 212 KB correct
10 Correct 0 ms 212 KB correct
11 Correct 1 ms 212 KB correct
12 Correct 0 ms 212 KB correct
13 Correct 1 ms 212 KB correct
14 Correct 6 ms 340 KB correct
15 Correct 5 ms 340 KB correct
16 Correct 5 ms 340 KB correct
17 Correct 5 ms 340 KB correct
18 Correct 2 ms 340 KB correct
19 Correct 5 ms 340 KB correct
20 Correct 5 ms 340 KB correct
21 Correct 4 ms 340 KB correct
22 Correct 4 ms 340 KB correct
23 Correct 3 ms 340 KB correct
24 Correct 3 ms 340 KB correct
25 Correct 1 ms 212 KB correct
26 Correct 3 ms 340 KB correct
27 Correct 3 ms 340 KB correct
28 Correct 2 ms 340 KB correct
29 Correct 2 ms 212 KB correct
30 Correct 3 ms 340 KB correct
31 Correct 3 ms 340 KB correct
32 Correct 3 ms 340 KB correct
33 Correct 3 ms 340 KB correct
34 Incorrect 239 ms 1492 KB WA in grader: NO
35 Halted 0 ms 0 KB -

Subtask #4
0 / 19.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 212 KB correct
2 Correct 0 ms 212 KB correct
3 Incorrect 166 ms 3796 KB WA in grader: NO
4 Halted 0 ms 0 KB -

Subtask #5
0 / 30.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 212 KB correct
2 Correct 0 ms 212 KB correct
3 Correct 1 ms 336 KB correct
4 Correct 0 ms 212 KB correct
5 Correct 0 ms 212 KB correct
6 Correct 1 ms 212 KB correct
7 Correct 1 ms 212 KB correct
8 Correct 0 ms 212 KB correct
9 Correct 0 ms 212 KB correct
10 Correct 0 ms 212 KB correct
11 Correct 1 ms 212 KB correct
12 Correct 0 ms 212 KB correct
13 Correct 1 ms 212 KB correct
14 Correct 6 ms 340 KB correct
15 Correct 5 ms 340 KB correct
16 Correct 5 ms 340 KB correct
17 Correct 5 ms 340 KB correct
18 Correct 2 ms 340 KB correct
19 Correct 5 ms 340 KB correct
20 Correct 5 ms 340 KB correct
21 Correct 4 ms 340 KB correct
22 Correct 4 ms 340 KB correct
23 Correct 3 ms 340 KB correct
24 Correct 3 ms 340 KB correct
25 Correct 1 ms 212 KB correct
26 Correct 3 ms 340 KB correct
27 Correct 3 ms 340 KB correct
28 Correct 2 ms 340 KB correct
29 Correct 2 ms 212 KB correct
30 Correct 3 ms 340 KB correct
31 Correct 3 ms 340 KB correct
32 Correct 3 ms 340 KB correct
33 Correct 3 ms 340 KB correct
34 Incorrect 239 ms 1492 KB WA in grader: NO
35 Halted 0 ms 0 KB -