Submission #253844

# Submission time Handle Problem Language Result Execution time Memory
253844 2020-07-28T22:09:13 Z aZvezda Pipes (BOI13_pipes) C++14
44.0741 / 100
1000 ms 32888 KB
#include <bits/stdc++.h>
using namespace std;
//#pragma GCC optimize ("O3")
//#pragma GCC target ("sse4")
#define endl "\n"
typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
template<class T, class T2> inline bool chkmax(T &x, const T2 &y) { return x < y ? x = y, 1 : 0; }
template<class T, class T2> inline bool chkmin(T &x, const T2 &y) { return x > y ? x = y, 1 : 0; }
const ll mod = 1e9 + 7;
template<class T> inline void fix(T &x) {if(x >= mod | x <= -mod) {x %= mod;} if(x < 0) {x += mod;}}
#define out(x) cout << __LINE__ << ": " << (#x) << " = " << (x) << endl


//If we have an even cycle then we can add 1 on odd edges and -1 on even edges so the solutions stay the same. We can remove leaves one by one

const ll MAX_N = 1e5 + 10;
set<pair<ll, ll> > g[MAX_N];
ll ans[MAX_N], arr[MAX_N];

struct cmp {
    bool operator ()(const ll &a, const ll &b) const {
        if(g[a].size() == g[b].size()) {
            return a < b;
        }
        return g[a].size() < g[b].size();
    }
};

multiset<ll, cmp> pq;

void removeBranches() {
    while(!pq.empty() && g[*(pq.begin())].size() == 1) {
        ll curr = *(pq.begin()); pq.erase(pq.begin());
        auto other = *(g[curr].begin());
        pq.erase(other.first);
        ans[other.second] = 2 * arr[curr];
        arr[other.first] -= arr[curr];
        g[other.first].erase({curr, other.second});
        pq.insert(other.first);
    }
}

int rret = 0;

void dfs(int x, int p, int start, int flag) {
    rret += arr[x] * flag;
    cout << x << " " << p << " " << ans[x] << " " << rret << endl;
    for(auto it : g[x]) {
        if(it.first == p || it.first == start) {continue;}
        dfs(it.first, x, start, -flag);
    }
}

void removeCycle() {
    if(pq.empty()) {return;}
    auto now = pq.begin();
    if(g[*now].size() > 2) {
        cout << 0 << endl;
        exit(0);
    }
    now = pq.end(); now --;
    if(g[*now].size() > 2) {
        cout << 0 << endl;
        exit(0);
    }
    if(pq.size() % 2 == 0) {
        cout << 0 << endl;
        exit(0);
    }
    int curr = *(pq.begin());
    dfs(curr, (*(g[curr].begin())).first, curr, 1);
    rret /= 2;
    auto sec = (g[curr].begin());
    arr[curr] -= rret;
    arr[(*sec).first] -= rret;
    ans[(*sec).second] = rret;
    g[curr].erase(sec);
    g[(*sec).first].erase({curr, (*sec).second});
    //cout << curr << " " << (*sec).first << " " << rret << endl;
}

signed main() {
    //ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);
    ll n, m;
    cin >> n >> m;
    if(m > n) {cout << 0 << endl; return 0;}
    for(ll i = 1; i <= n; i ++) {
        cin >> arr[i];
    }
    for(ll i = 0; i < m; i ++) {
        ll a, b;
        cin >> a >> b;
        g[a].insert({b, i});
        g[b].insert({a, i});
    }
    for(ll i = 1; i <= n; i ++) {
        pq.insert(i);
    }
    removeBranches();
    removeCycle();
    removeBranches();
    for(ll i = 0; i < m; i ++) {
        cout << ans[i] << endl;
    }
    return 0;
}


# Verdict Execution time Memory Grader output
1 Execution timed out 1099 ms 4992 KB Time limit exceeded
2 Execution timed out 1099 ms 4992 KB Time limit exceeded
3 Execution timed out 1099 ms 5248 KB Time limit exceeded
4 Execution timed out 1100 ms 23928 KB Time limit exceeded
5 Execution timed out 1099 ms 4992 KB Time limit exceeded
6 Execution timed out 1089 ms 4992 KB Time limit exceeded
7 Execution timed out 1095 ms 5120 KB Time limit exceeded
8 Execution timed out 1099 ms 4992 KB Time limit exceeded
9 Execution timed out 1099 ms 5248 KB Time limit exceeded
10 Execution timed out 1099 ms 5248 KB Time limit exceeded
11 Execution timed out 1087 ms 5248 KB Time limit exceeded
12 Execution timed out 1088 ms 5120 KB Time limit exceeded
13 Execution timed out 1074 ms 20088 KB Time limit exceeded
14 Execution timed out 1088 ms 22904 KB Time limit exceeded
15 Execution timed out 1100 ms 24056 KB Time limit exceeded
16 Execution timed out 1100 ms 21112 KB Time limit exceeded
17 Execution timed out 1100 ms 23928 KB Time limit exceeded
18 Execution timed out 1080 ms 24056 KB Time limit exceeded
19 Execution timed out 1090 ms 24056 KB Time limit exceeded
20 Execution timed out 1098 ms 4992 KB Time limit exceeded
21 Execution timed out 1099 ms 5248 KB Time limit exceeded
22 Execution timed out 1100 ms 24032 KB Time limit exceeded
23 Execution timed out 1053 ms 20088 KB Time limit exceeded
24 Execution timed out 1084 ms 24056 KB Time limit exceeded
25 Execution timed out 1096 ms 20856 KB Time limit exceeded
# Verdict Execution time Memory Grader output
1 Incorrect 3 ms 5120 KB Output isn't correct
2 Incorrect 5 ms 5248 KB Output isn't correct
3 Correct 239 ms 22904 KB Output is correct
4 Correct 3 ms 4992 KB Output is correct
5 Correct 3 ms 4992 KB Output is correct
6 Correct 3 ms 4992 KB Output is correct
7 Incorrect 3 ms 4992 KB Output isn't correct
8 Incorrect 3 ms 4992 KB Output isn't correct
9 Correct 3 ms 4992 KB Output is correct
10 Correct 3 ms 4992 KB Output is correct
11 Correct 3 ms 4992 KB Output is correct
12 Correct 4 ms 4992 KB Output is correct
13 Correct 3 ms 4992 KB Output is correct
14 Incorrect 3 ms 4992 KB Output isn't correct
15 Incorrect 8 ms 5248 KB Output isn't correct
16 Incorrect 5 ms 5248 KB Output isn't correct
17 Correct 5 ms 5248 KB Output is correct
18 Correct 3 ms 4992 KB Output is correct
19 Correct 3 ms 4992 KB Output is correct
20 Correct 3 ms 4992 KB Output is correct
21 Correct 3 ms 4992 KB Output is correct
22 Incorrect 5 ms 5248 KB Output isn't correct
23 Incorrect 280 ms 25848 KB Output isn't correct
24 Incorrect 363 ms 29560 KB Output isn't correct
25 Correct 237 ms 22904 KB Output is correct
26 Correct 3 ms 4992 KB Output is correct
27 Correct 3 ms 4992 KB Output is correct
28 Correct 3 ms 4992 KB Output is correct
29 Correct 3 ms 4992 KB Output is correct
30 Incorrect 286 ms 28536 KB Output isn't correct
31 Incorrect 330 ms 32888 KB Output isn't correct
32 Incorrect 327 ms 26104 KB Output isn't correct
33 Correct 237 ms 24016 KB Output is correct
34 Correct 3 ms 4992 KB Output is correct
35 Correct 3 ms 4992 KB Output is correct
36 Correct 3 ms 4992 KB Output is correct
37 Correct 3 ms 4992 KB Output is correct
38 Incorrect 306 ms 28920 KB Output isn't correct
39 Incorrect 314 ms 25360 KB Output isn't correct
40 Incorrect 322 ms 29176 KB Output isn't correct
41 Correct 189 ms 23928 KB Output is correct
42 Correct 3 ms 4992 KB Output is correct
43 Correct 3 ms 4992 KB Output is correct
44 Correct 3 ms 4992 KB Output is correct
45 Correct 3 ms 4992 KB Output is correct
46 Incorrect 289 ms 28408 KB Output isn't correct
47 Incorrect 330 ms 29264 KB Output isn't correct
48 Incorrect 327 ms 32504 KB Output isn't correct
49 Correct 279 ms 22904 KB Output is correct
50 Correct 3 ms 4992 KB Output is correct
51 Correct 3 ms 4992 KB Output is correct
52 Correct 3 ms 4992 KB Output is correct
53 Correct 3 ms 4992 KB Output is correct
54 Incorrect 314 ms 27768 KB Output isn't correct