oj.uz

답안 #846259

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
846259 2023-09-07T13:01:42 Z vjudge1 KOVANICE (COI15_kovanice) C++17
60 / 100
 2000 ms 30908 KB 
#include <bits/stdc++.h>
using namespace std;
 
#define fast ios::sync_with_stdio(0);cin.tie(0);
#define s second
#define f first
typedef long long ll;
const ll MOD = 998244353;
const ll LOGN = 20;
const ll MAXN = 3e5 + 5;

vector<vector<int>> graph, inv;
int par[MAXN], sz[MAXN], indeg[MAXN];
int dist[MAXN];
int find(int node) {
	if (node == par[node])
		return node;
	return par[node] = find(par[node]);
}

void unite(int a, int b) {
	a = find(a);
	b = find(b);
	if (a == b)
		return ;
	if (sz[b] > sz[a])
		swap(a, b);
	par[b] = a;
	sz[a] += sz[b];
}

vector<int> type;
void dfs(int node, int req) {
	node = find(node);
	if (type[node] != -1 || dist[node] + 1 != req || req == 0)
		return ;

	type[node] = req;
	for (auto u : inv[node]) {
		dfs(u, req - 1);
	}
}

void getDist(int node) {
	for (auto u : graph[node]) {
		if (dist[u] >= dist[node] + 1)
			continue;
		dist[u] = dist[node] + 1;
		getDist(u);
	}
}

vector<pair<int,int>> edge;
int main() {
	fast
	int N, M, V;
	cin >> N >> M >> V;

	graph = vector<vector<int>>(M+1, vector<int>());
	inv = vector<vector<int>>(M+1, vector<int>());
	type = vector<int>(M+1, -1);
	for (int i = 1; i <= M; i++)
		par[i] = i, sz[i] = 1;

	for (int i = 0; i < V; i++) {
		string s;
		cin >> s;

		vector<int> v1, v2;
		int plc = -1;
		for (int i = 0; i < s.length(); i++) {
			if (s[i] != '=' && s[i] != '<')
				v1.push_back(s[i] - '0');
			else {
				plc = i;
				break;
			}
		}
		for (int i = plc + 1; i < s.length(); i++) {
			v2.push_back(s[i] - '0');
		}

		int now = 1;
		int l = 0;
		int r = 0;
		while (v1.size()) {
			l += now * v1.back();
			now *= 10;
			v1.pop_back(); 
		}
		now = 1;
		while (v2.size()) {
			r += now * v2.back();
			now *= 10;
			v2.pop_back();
		}

		if (s[plc] == '=')
			unite(l, r);
		else
			edge.push_back({l, r});
	}

	for (int i = 0; i < edge.size(); i++) {
		edge[i].f = find(edge[i].f);
		edge[i].s = find(edge[i].s);
		graph[edge[i].f].push_back(edge[i].s); 
		inv[edge[i].s].push_back(edge[i].f);
		indeg[edge[i].s]++;
	}

	for (int i = 1; i <= M; i++) {
		if (i == find(i) && indeg[i] == 0)
			getDist(i);
	}

	for (int i = 1; i <= M; i++) {
		if (dist[i] + 1 == N)
			dfs(i, N);
	}

	for (int i = 1; i <= M; i++) {
		if (type[find(i)] == -1)
			cout << "?" << endl;
		else
			cout << "K" << type[find(i)] << endl;
	}
}

Compilation message

kovanice.cpp: In function 'int main()':
kovanice.cpp:71:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   71 |   for (int i = 0; i < s.length(); i++) {
      |                   ~~^~~~~~~~~~~~
kovanice.cpp:79:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   79 |   for (int i = plc + 1; i < s.length(); i++) {
      |                         ~~^~~~~~~~~~~~
kovanice.cpp:104:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  104 |  for (int i = 0; i < edge.size(); i++) {
      |                  ~~^~~~~~~~~~~~~

Subtask #1
10.0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Correct 3 ms 4440 KB Output is correct
2 Correct 3 ms 4440 KB Output is correct

Subtask #2
40.0 / 40.0

# 결과 실행 시간 메모리 Grader output
1 Correct 268 ms 18384 KB Output is correct
2 Correct 269 ms 18636 KB Output is correct

Subtask #3
10.0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Correct 23 ms 6340 KB Output is correct
2 Correct 24 ms 6884 KB Output is correct
3 Correct 33 ms 6344 KB Output is correct

Subtask #4
0 / 40.0

# 결과 실행 시간 메모리 Grader output
1 Correct 561 ms 30908 KB Output is correct
2 Execution timed out 2029 ms 30548 KB Time limit exceeded
3 Halted 0 ms 0 KB -