oj.uz

Submission #1095288

# Submission time Handle Problem Language Result Execution time Memory
1095288 2024-10-01T18:31:42 Z andrewp Paths (BOI18_paths) C++14
0 / 100
 126 ms 31236 KB 
//Dedicated to my love, ivaziva
#pragma GCC optimize("Ofast") 
#include <bits/stdc++.h> 
using namespace std;  

#define int long long 
 
using pii = pair<int, int>;
using ll = int64_t; 
 
#define all(x) x.begin(), x.end()
#define rall(x) x.rbegin(), x.rend()
#define dbg(x) cerr<<#x<<": "<<x<<'\n';  
#define dbga(A,l_,r_){for(int i_=l_;i_<=r_;i_++)cerr<<A[i_]<<' ';cerr<<'\n';}
#define dbgv(a_){for(auto x_:a_) cerr<<x_<<' ';cerr<<'\n';}   

const int N = 3e5 + 20;
int n, m, k, a[N], ans1, ans2;
vector<int> g[N];
bool was[N];

void dfs(int x) {
    was[x] = true; 
    int f = 0, s = 0, t = 0;
    for (int u : g[x]) {
        if (a[u] == 1) {
            f++;
        } else if (a[u] == 2) {
            s++;
        } else {
            t++;
        }
        if (!was[u]) { 
            dfs(u);
        }
    }   
    if (a[x] == 1) {
        ans1 += 2 * (s * t); 
        ans2 += s;  
        ans2 += t; 
    } else if (a[x] == 2) {
        ans1 += 2 * (f * t); 
        ans2 += f;
        ans2 += t;  
    } else { 
        ans1 += 2 * (f * s);  
        ans2 += f; 
        ans2 += s; 
    }  
}

int32_t main()  {
    ios::sync_with_stdio(false); cin.tie(nullptr);  
    cout.tie(nullptr); cerr.tie(nullptr);

    cin >> n >> m >> k;
    for (int i = 1; i <= n; i++) {
        cin >> a[i];
        was[i] = false;
    }
    for (int i = 0; i < m; i++) {
        int u, v;
        cin >> u >> v;
        g[u].push_back(v);
        g[v].push_back(u);
    }
    ans1 = 0, ans2 = 0;
    dfs(1);    
    if (k == 1) {
        cout << 0 << '\n';
    } else if (k == 2) {
        cout << ans1 << '\n'; 
    } else if (k == 3) {
        cout << ans1 + ans2 << '\n';
    } 
    return 0;
}

Subtask #1
0 / 23.0

# Verdict Execution time Memory Grader output
1 Incorrect 4 ms 7256 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #2
0 / 20.0

# Verdict Execution time Memory Grader output
1 Correct 46 ms 17828 KB Output is correct
2 Correct 37 ms 16720 KB Output is correct
3 Incorrect 126 ms 31236 KB Output isn't correct
4 Halted 0 ms 0 KB -

Subtask #3
0 / 27.0

# Verdict Execution time Memory Grader output
1 Incorrect 4 ms 7256 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #4
0 / 30.0

# Verdict Execution time Memory Grader output
1 Incorrect 4 ms 7260 KB Output isn't correct
2 Halted 0 ms 0 KB -