Submission #1030442

#	Time	Username	Problem	Language	Result	Execution time	Memory
1030442		thinknoexit	Graph (BOI20_graph)	C++17	0 / 100	2 ms	2812 KiB

Graph

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
const int N = 100100;
struct A {
    int v, w;
};
vector<A> adj[N];
double ans[N];
bool ch[N], vis[N];
int col[N], bp = 0;
vector<pair<int, int>> cycle;
void dfs(int v, vector<int>& node) {
    node.push_back(v);
    vis[v] = 1;
    for (auto& x : adj[v]) {
        if (!vis[x.v]) dfs(x.v, node);
    }
}
void dfs_ans(int v) {
    for (auto& x : adj[v]) {
        if (!ch[x.v]) {
            ch[x.v] = 1;
            ans[x.v] = x.w - ans[v];
            dfs_ans(x.v);
        }
    }
}
bool bipartite(int v, int c) {
    if (col[v]) {
        if (col[v] != c) return bp = v, false;
        else return true;
    }
    col[v] = c;
    for (auto& x : adj[v]) {
        if (!bipartite(x.v, 3 - c)) {
            if (bp != 0) cycle.push_back({ v, x.w });
            if (v == bp) bp = 0;
            return false;
        }
    }
    return true;
}
int main() {
    cin.tie(nullptr)->sync_with_stdio(false);
    int n, m;
    cin >> n >> m;
    for (int i = 1;i <= m;i++) {
        int u, v, w;
        cin >> u >> v >> w;
        adj[u].push_back({ v, w });
        adj[v].push_back({ u, w });
    }
    for (int i = 1;i <= n;i++) {
        if (vis[i]) continue;
        vector<int> node;
        dfs(i, node);
        bool bipar = bipartite(i, 1);
        if (!bipar) {
            vector<pair<int, int>> c = cycle;
            cycle.clear();
            int root = c.back().first;
            int val = 0;
            for (int i = 0;i < (int)c.size();i++) {
                if (i & 1) val -= c[i].second;
                else val += c[i].second;
            }
            // value at node (root) = val / 2
            ch[root] = 1;
            ans[root] = (double)val / 2;
            dfs_ans(root);
            continue;
        }
        if ((int)node.size() == 1) {
            continue;
        }
        vector<int> c[2];
        for (auto& x : node) {
            c[col[x] - 1].push_back(x);
        }
        int mn = 1e9;
        int idx = -1;
        for (int j = 0;j < 2;j++) {
            int sum = 0;
            bool ch2 = 1;
            for (auto& x : c[j]) {
                int now = adj[x][0].w;
                sum += now;
                for (auto& y : adj[x]) {
                    if (y.w != now) ch2 = 0;
                }
            }
            if (!ch2) continue;
            if (sum < mn) {
                mn = sum;
                idx = j;
            }
        }
        if (mn == 1e9) {
            cout << "NO";
            return 0;
        }
        for (auto& x : node) ch[x] = 1;
        for (auto& x : c[idx]) {
            ans[x] = adj[x][0].w;
        }
    }
    for (int i = 1;i <= n;i++) {
        for (auto& x : adj[i]) {
            if (ans[i] + ans[x.v] != x.w) {
                cout << "NO";
                return 0;
            }
        }
    }
    cout << "YES\n";
    for (int i = 1;i <= n;i++) {
        cout << ans[i] << ' ';
    }
    return 0;
}

• Subtask 1: 0 / 5
• Subtask 2: 0 / 12
• Subtask 3: 0 / 17
• Subtask 4: 0 / 24
• Subtask 5: 0 / 42