이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
using namespace std;
#define mp make_pair
#define pb push_back
#define fi first
#define se second
int n, m, cnt[100007], vis[500007], u, v, elim = 0, cur = 0;
long long c[100007], ans[500007], sum = 0;
vector < pair <int, int> > edge[100007];
void solve(){
queue <int> q;
memset(cnt, 0, sizeof(cnt));
for(int i = 1; i <= n; i++){
for(int j = 0; j < (int)edge[i].size(); j++){
if(edge[i][j].se == elim) continue;
cnt[i]++;
}
if(cnt[i] == 1) q.push(i);
}
vis[elim] = true;
while(!q.empty()){
int u = q.front();
cnt[u]--;
q.pop();
for(int i = 0; i < (int)edge[u].size(); i++){
int v = edge[u][i].fi, id = edge[u][i].se;
//cout << u << " " << v << " " << id << endl;
if(vis[id]) continue;
ans[id] = c[u] * 2LL;
c[u] -= ans[id] / 2LL, c[v] -= ans[id] / 2LL;
cnt[v]--;
vis[id] = true;
if(cnt[v] == 1) q.push(v);
}
}
for(int i = 1; i <= m; i++) cout << ans[i] << "\n";
exit(0);
}
void dfs(int u, int par){
cur++;
if(cur & 1) sum += c[u];
else sum -= c[u];
for(int i = 0; i < (int)edge[u].size(); i++){
int v = edge[u][i].fi, id = edge[u][i].se;
if(v == par) continue;
if(cur == n){
ans[id] = sum;
c[u] -= sum / 2LL;
c[v] -= sum / 2LL;
elim = id;
solve();
return;
}
dfs(v, u);
}
}
int main(){
ios_base::sync_with_stdio(0);
cin >> n >> m;
for(int i = 1; i <= n; i++){
cin >> c[i];
}
for(int i = 1; i <= m; i++){
cin >> u >> v;
edge[u].pb(mp(v, i));
edge[v].pb(mp(u, i));
}
if(m > n) return cout << 0, 0;
if(m == n - 1) solve();
if(n % 2 == 0) return cout << 0, 0;
dfs(1, 1);
solve();
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |