Submission #2095929

---

Source Code Expand

#include "bits/stdc++.h"
using namespace std;
const long long mod = 1000000007;
const int maxn = 100100;
vector<int> tr[maxn];
int a[maxn], b[maxn], c[maxn], sz[maxn];
long long s[maxn], dw[maxn], res[maxn];
int n;
void solve(int cur, int pr = -1){
    sz[cur] = 1;
    dw[cur] = 0;
    for(int id : tr[cur]){
        int nxt = cur ^ c[id];
        if(nxt == pr) continue;
        solve(nxt, cur);
        sz[cur] += sz[nxt];
    }
    for(int id : tr[cur]){
        int nxt = cur ^ c[id];
        if(nxt == pr) continue;
        int den = ((n - sz[nxt] + 1) - 1 - sz[nxt]);
        if(den == 0){
            assert(0);
            res[id] = (s[cur] - dw[nxt]) / (sz[nxt]);
        } else {
            res[id] = (s[nxt] - s[cur]) / den;
        }
        dw[cur] += dw[nxt] + (sz[nxt]) * res[id];
    }
}
int main(){
    scanf("%d", &n);
    for(int e = 0; e < n - 1; e++){
        scanf("%d %d", a + e, b + e); c[e] = a[e] ^ b[e];
        tr[a[e]].push_back(e);
        tr[b[e]].push_back(e);
    }
    for(int e = 1; e <= n; e++) scanf("%lld", s + e);
    solve(1);
    for(int e = 0; e < n - 1; e++) cout << res[e] << endl;
	return 0;
}

Submission Info

Compile Error

./Main.cpp: In function ‘int main()’:
./Main.cpp:32:20: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
     scanf("%d", &n);
                    ^
./Main.cpp:34:37: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
         scanf("%d %d", a + e, b + e); c[e] = a[e] ^ b[e];
                                     ^
./Main.cpp:38:53: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
     for(int e = 1; e <= n; e++) scanf("%lld", s + e);
                                                     ^

Judge Result