oj.uz

Submission #637221

# Submission time Handle Problem Language Result Execution time Memory
637221 2022-08-31T23:55:02 Z MinaRagy06 KOVANICE (COI15_kovanice) C++17
50 / 100
 2000 ms 46880 KB 
#include <bits/stdc++.h>
using namespace std;
#define lesgooo ios_base::sync_with_stdio(0), cin.tie(0), cout.tie(0)
#define endl    '\n'
#define int     long long
 
const int N = 3e5+5;
vector<int> adj[N], comp[N];
int idx[N], vis[N], cur, indeg[N], val[N], ans[N], dp[N], vis2[N];
void dfs(int i)
{
    comp[cur].push_back(i), idx[i] = cur, vis[i] =  1;
    for (auto nxt : adj[i]) if (!vis[nxt]) dfs(nxt);
}
int depth(int i)
{
    if (~dp[i]) return dp[i];
    dp[i] = 1;
    for (auto nxt : adj[i]) dp[i] = max(dp[i], 1+depth(nxt));
    return dp[i];
}
void assign(int i, int dep)
{
    if (!vis[i]) return;
    val[i] = dep, vis[i] = 1;
    for (auto nxt : adj[i]) assign(nxt, dep+1);
}
signed main()
{
    lesgooo;
    int n, m, v;
    cin >> n >> m >> v;
    pair<pair<int, char>, int> a[v];
    for (int i = 0; i < v; i++)
    {
        cin >> a[i].first.first >> a[i].first.second >> a[i].second;
        a[i].first.first--, a[i].second--;
        if (a[i].first.second == '=') adj[a[i].first.first].push_back(a[i].second), adj[a[i].second].push_back(a[i].first.first);
    }
    for (int i = 0; i < m; i++) if (!vis[i]) dfs(i), cur++;
    for (int i = 0; i < m; i++) adj[i].clear(), val[i] = -1;
    for (int i = 0; i < v; i++) if (a[i].first.second != '=') adj[idx[a[i].first.first]].push_back(idx[a[i].second]), indeg[idx[a[i].second]] = 1;
    // for (int i = 0; i < cur; i++)
    // {
    //     cout << i << ": ";
    //     for (auto nxt : adj[i]) cout << nxt << " ";
    //     cout << endl;
    // }
    memset(dp, -1, sizeof dp);
    for (int i = 0; i < cur; i++) if (!indeg[i] && depth(i) == n) assign(i, 1);
    for (int i = 0; i < m; i++) ans[i] = val[idx[i]];
    for (int i = 0; i < m; i++) cout << (ans[i] == -1? "?" : "K"+to_string(ans[i])) << endl;
    return 0;
}

Subtask #1
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 9 ms 16888 KB Output is correct
2 Correct 10 ms 16948 KB Output is correct

Subtask #2
40.0 / 40.0

# Verdict Execution time Memory Grader output
1 Correct 102 ms 35276 KB Output is correct
2 Correct 94 ms 35348 KB Output is correct

Subtask #3
0 / 10.0

# Verdict Execution time Memory Grader output
1 Execution timed out 2090 ms 20428 KB Time limit exceeded
2 Halted 0 ms 0 KB -

Subtask #4
0 / 40.0

# Verdict Execution time Memory Grader output
1 Execution timed out 2082 ms 46880 KB Time limit exceeded
2 Halted 0 ms 0 KB -