Submission #1126181

#TimeUsernameProblemLanguageResultExecution timeMemory
1126181MuhammetPipes (BOI13_pipes)C++20
65 / 100
132 ms15196 KiB
#include <bits/stdc++.h>

using namespace std;

#define SZ(s) (int)s.size()

int n, m;

map <pair<int,int>,int> ans;

vector <int> a, u1, u2, deg;

vector <vector <int>> v;

int main(){
	ios::sync_with_stdio(false); cin.tie(nullptr);

	cin >> n >> m;
	a.resize(n+1);
	for(int i = 1; i <= n; i++){
		cin >> a[i];
	}
	v.resize(n+1), deg.resize(n+1,0);
	u1.resize(m+1), u2.resize(m+1);
	for(int i = 1; i <= m; i++){
		cin >> u1[i] >> u2[i];
		v[u1[i]].push_back(u2[i]);
		v[u2[i]].push_back(u1[i]);
	}
	if(m > n){
		cout << 0;
		return 0;
	}
	queue <int> q;
	for(int i = 1; i <= n; i++){
		deg[i] = SZ(v[i]);
		if(deg[i] == 1){
			q.push(i);
		}
	}
	while(!q.empty()){
		int ind = q.front();
		q.pop();
		deg[ind]--;
		for(auto i : v[ind]){
			if(deg[i] > 0){
				deg[i]--;
				ans[{min(i,ind), max(i,ind)}] = a[ind];
				a[i] -= a[ind];
				if(deg[i] == 1) q.push(i);
			}
		}
	}
	if(n == m+1){
		for(int i = 1; i <= m; i++){
			cout << 2*ans[{min(u1[i], u2[i]), max(u1[i], u2[i])}] << '\n';
		}
		return 0;
	}
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...