Submission #1872915
Source Code Expand
#include <bits/stdc++.h>
#define ll long long
#define INF 1000000005
#define MOD 1000000007
#define EPS 1e-10
#define rep(i,n) for(int i=0;i<(int)(n);++i)
#define rrep(i,n) for(int i=(int)(n)-1;i>=0;--i)
#define srep(i,s,t) for(int i=(int)(s);i<(int)(t);++i)
#define each(a,b) for(auto (a): (b))
#define all(v) (v).begin(),(v).end()
#define len(v) (int)(v).size()
#define zip(v) sort(all(v)),v.erase(unique(all(v)),v.end())
#define cmx(x,y) x=max(x,y)
#define cmn(x,y) x=min(x,y)
#define fi first
#define se second
#define pb push_back
#define show(x) cout<<#x<<" = "<<(x)<<endl
#define spair(p) cout<<#p<<": "<<p.fi<<" "<<p.se<<endl
#define svec(v) cout<<#v<<":";rep(kbrni,v.size())cout<<" "<<v[kbrni];cout<<endl
#define sset(s) cout<<#s<<":";each(kbrni,s)cout<<" "<<kbrni;cout<<endl
#define smap(m) cout<<#m<<":";each(kbrni,m)cout<<" {"<<kbrni.first<<":"<<kbrni.second<<"}";cout<<endl
using namespace std;
typedef pair<int,int> P;
typedef pair<ll,ll> pll;
typedef vector<int> vi;
typedef vector<vi> vvi;
typedef vector<ll> vl;
typedef vector<vl> vvl;
typedef vector<double> vd;
typedef vector<P> vp;
typedef vector<string> vs;
const int MAX_N = 100005;
struct edge
{
int to;
ll cost;
};
vector<int> G[MAX_N];
vector<edge> graph[MAX_N];
int par[MAX_N];
int st[MAX_N];
ll sma;
void dfs(int u,int p)
{
par[u] = p;
st[u] = 1;
rep(i,len(G[u])){
if(G[u][i] != p){
dfs(G[u][i],u);
st[u] += st[G[u][i]];
}
}
}
void redfs(int u,int p,ll dist)
{
sma += dist;
rep(i,len(graph[u])){
if(graph[u][i].to != p){
redfs(graph[u][i].to,u,dist+graph[u][i].cost);
}
}
}
int main()
{
cin.tie(0);
ios::sync_with_stdio(false);
int n;
cin >> n;
vp vec(n-1);
rep(i,n-1){
int a,b;
cin >> a >> b;
--a,--b;
vec[i] = P(a,b);
G[a].pb(b),G[b].pb(a);
}
vl sm(n);
rep(i,n){
cin >> sm[i];
}
dfs(0,-1);
vl ans(n-1);
int flag = -1;
bool hoge = false;
rep(i,n-1){
int a = vec[i].fi,b = vec[i].se;
if(par[a] != b){
hoge = true;
swap(a,b);
}
if(n == 2*st[a]){
flag = i;
continue;
}
ans[i] = (sm[a]-sm[b])/(n-2*st[a]);
}
if(flag >= 0){
rep(i,n-1){
int a = vec[i].fi,b = vec[i].se;
if(flag == i){
graph[a].pb((edge){b,0}),graph[b].pb((edge){a,0});
}else{
graph[a].pb((edge){b,ans[i]}),graph[b].pb((edge){a,ans[i]});
}
}
sma = 0;
if(hoge){
redfs(vec[flag].fi,-1,0);
ans[flag] = (sm[vec[flag].fi]-sma) / st[vec[flag].se];
}else{
redfs(vec[flag].se,-1,0);
ans[flag] = (sm[vec[flag].se]-sma) / st[vec[flag].fi];
}
}
rep(i,n-1){
cout << ans[i] << "\n";
}
return 0;
}
Submission Info
Submission Time |
|
Task |
D - Ancient Tree Record |
User |
kopricky |
Language |
C++14 (GCC 5.4.1) |
Score |
500 |
Code Size |
3074 Byte |
Status |
WA |
Exec Time |
79 ms |
Memory |
24320 KB |
Judge Result
Set Name |
Sample |
Subtask1 |
Subtask2 |
All |
Score / Max Score |
0 / 0 |
300 / 300 |
200 / 200 |
0 / 300 |
Status |
|
|
|
|
Set Name |
Test Cases |
Sample |
00_example_01.txt, 00_example_02.txt, 00_example_03.txt |
Subtask1 |
00_example_01.txt, s1_01.txt, s1_02.txt, s1_03.txt, s1_04.txt, s1_05.txt, s1_06.txt, s1_07.txt, s1_08.txt, s1_09.txt, s1_10.txt, s1_11.txt, s1_12.txt, s1_13.txt |
Subtask2 |
00_example_02.txt, s2_14.txt, s2_15.txt, s2_16.txt, s2_17.txt, s2_18.txt, s2_19.txt, s2_20.txt, s2_21.txt, s2_22.txt, s2_23.txt |
All |
00_example_01.txt, 00_example_02.txt, 00_example_03.txt, s1_01.txt, s1_02.txt, s1_03.txt, s1_04.txt, s1_05.txt, s1_06.txt, s1_07.txt, s1_08.txt, s1_09.txt, s1_10.txt, s1_11.txt, s1_12.txt, s1_13.txt, s2_14.txt, s2_15.txt, s2_16.txt, s2_17.txt, s2_18.txt, s2_19.txt, s2_20.txt, s2_21.txt, s2_22.txt, s2_23.txt, s3_24.txt, s3_25.txt, s3_26.txt, s3_27.txt, s3_28.txt, s3_29.txt, s3_30.txt, s3_31.txt, s3_32.txt, s3_33.txt, s3_34.txt, s3_35.txt, s3_36.txt, s3_37.txt, s3_38.txt, s3_39.txt, s3_40.txt, s3_41.txt, s3_42.txt, s3_43.txt, s3_44.txt, s3_45.txt |
Case Name |
Status |
Exec Time |
Memory |
00_example_01.txt |
AC |
3 ms |
4992 KB |
00_example_02.txt |
AC |
3 ms |
4992 KB |
00_example_03.txt |
AC |
3 ms |
4992 KB |
s1_01.txt |
AC |
40 ms |
15872 KB |
s1_02.txt |
AC |
47 ms |
16640 KB |
s1_03.txt |
AC |
5 ms |
5632 KB |
s1_04.txt |
AC |
39 ms |
14592 KB |
s1_05.txt |
AC |
9 ms |
6528 KB |
s1_06.txt |
AC |
58 ms |
19584 KB |
s1_07.txt |
AC |
69 ms |
24320 KB |
s1_08.txt |
AC |
59 ms |
19584 KB |
s1_09.txt |
AC |
3 ms |
4992 KB |
s1_10.txt |
AC |
3 ms |
4992 KB |
s1_11.txt |
AC |
3 ms |
4992 KB |
s1_12.txt |
AC |
3 ms |
4992 KB |
s1_13.txt |
AC |
3 ms |
4992 KB |
s2_14.txt |
AC |
28 ms |
9084 KB |
s2_15.txt |
AC |
38 ms |
10872 KB |
s2_16.txt |
AC |
48 ms |
12280 KB |
s2_17.txt |
AC |
47 ms |
12280 KB |
s2_18.txt |
AC |
47 ms |
12280 KB |
s2_19.txt |
AC |
48 ms |
12280 KB |
s2_20.txt |
AC |
3 ms |
4992 KB |
s2_21.txt |
AC |
3 ms |
4992 KB |
s2_22.txt |
AC |
3 ms |
4992 KB |
s2_23.txt |
AC |
3 ms |
4992 KB |
s3_24.txt |
AC |
33 ms |
8960 KB |
s3_25.txt |
AC |
48 ms |
10496 KB |
s3_26.txt |
AC |
5 ms |
5248 KB |
s3_27.txt |
AC |
40 ms |
9472 KB |
s3_28.txt |
AC |
9 ms |
5760 KB |
s3_29.txt |
AC |
60 ms |
11904 KB |
s3_30.txt |
AC |
59 ms |
12032 KB |
s3_31.txt |
AC |
60 ms |
11904 KB |
s3_32.txt |
AC |
76 ms |
20736 KB |
s3_33.txt |
AC |
63 ms |
18432 KB |
s3_34.txt |
AC |
53 ms |
12280 KB |
s3_35.txt |
AC |
54 ms |
12280 KB |
s3_36.txt |
WA |
3 ms |
4992 KB |
s3_37.txt |
AC |
3 ms |
4992 KB |
s3_38.txt |
AC |
3 ms |
4992 KB |
s3_39.txt |
AC |
3 ms |
4992 KB |
s3_40.txt |
AC |
42 ms |
11904 KB |
s3_41.txt |
WA |
62 ms |
14464 KB |
s3_42.txt |
WA |
5 ms |
5376 KB |
s3_43.txt |
WA |
79 ms |
17024 KB |
s3_44.txt |
WA |
76 ms |
17280 KB |
s3_45.txt |
WA |
79 ms |
17024 KB |