#include <bits/stdc++.h>
using namespace std;
struct dsu{
int len;
vector<int> parent;
dsu(int ln): len(ln), parent(ln,-1) {
for(int i=0; i<ln; i++)
parent[i] = i;
}
void connect(int a, int b){
a = (*this)[a];
b = (*this)[b];
if(a>b)
a^=b^=a^=b;
parent[b] = a;
}
int operator[](int x) {
if( parent[x] != x )
return parent[x] = (*this)[parent[x]];
return x;
}
};
int main(){
int n,m,v;
cin >> n >> m >> v;
vector<pair<int,int>> less;
dsu equal(m), grup(m);
while(v--){
int a,b;
char c;
cin >> a >> c >> b;
a--;
b--;
grup.connect(a,b);
if(c == '='){
equal.connect(a,b);
}else{
less.push_back({a,b});
}
}
vector<int> type(m, -1), root(m, 1);
vector<vector<int>> graph(m);
for(auto [a, b]: less){
a = equal[a];
b = equal[b];
assert(a != b);
root[a] = 0;
graph[b].push_back(a);
}
vector<int> longest_path_below(m, -1);
function<int(int)> dfs = [&](int node){
int &lpb = longest_path_below[node];
if(lpb != -1)
return lpb;
lpb = 0;
for(int i=0; i<graph[node].size(); i++)
lpb = max(lpb, dfs(graph[node][i])+1);
return lpb;
};
for(int i=0; i<m; i++)
dfs(equal[i]);
function< void(int, int) > settype = [&](int node, int tip){
if( longest_path_below[node] +1 != tip )
return;
if( type[node] != -1 )
return;
type[node] = tip;
for(int i=0; i<graph[node].size(); i++)
settype( graph[node][i], tip-1 );
};
for(int i=0; i<m; i++)
settype(i, n);
for(int i=0; i<m; i++){
int t = type[equal[i]];
// cout << longest_path_below[equal[i]] << " ";
if(t==-1)
cout << "K?\n";
else
cout << "K" << t << "\n";
}
}
Compilation message
kovanice.cpp: In lambda function:
kovanice.cpp:86:17: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
86 | for(int i=0; i<graph[node].size(); i++)
| ~^~~~~~~~~~~~~~~~~~~
kovanice.cpp: In lambda function:
kovanice.cpp:103:17: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
103 | for(int i=0; i<graph[node].size(); i++)
| ~^~~~~~~~~~~~~~~~~~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
2 ms |
344 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
88 ms |
11208 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
37 ms |
1732 KB |
Output is correct |
2 |
Correct |
36 ms |
1692 KB |
Output is correct |
3 |
Correct |
36 ms |
1728 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
217 ms |
20272 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |