【PA2014】Final Zadanie

Source

PA2014
bzoj5481

Hint

请先思考后再展开

众所周知这个C可以二次扫描求出来,形式为Ci=xi+yi
但现在是已知C求a,上面这东西我不知道怎么求
此时不妨换个角度,考虑从x移动到fa,C有怎样的变化,列一些方程

Solution

请先思考后再展开

设tot为子树人数和
$2tot_x=tot_1+C_{fa}-C_x$
此时只有n-1个方程,玩玩样例发现不对,发现忽略了一个基本方程
$C_1=\sum (tot_x-\sum tot_{son})dis(x,1)=\sum_{x>1} tot_x$
于是可以直接求出 $tot_1$
于是其他tot也可以直接求了

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
//Zory-2019
#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<map>
#include<set>
#include<queue>
#include<deque>
#include<stack>
#include<bitset>
#include<vector>
#include<algorithm>
#include<iostream>
#include<deque>
using namespace std;
namespace mine
{
typedef long long ll;
#define double long double
const int INF=0x3f3f3f3f;
const ll LLINF=0x3f3f3f3f3f3f3f3fll;
ll qread()
{
ll ans=0;char c=getchar();int f=1;
while(c<'0' or c>'9') {if(c=='-') f=-1;c=getchar();}
while('0'<=c and c<='9') ans=ans*10+c-'0',c=getchar();
return ans*f;
}
void write(ll num)
{
if(num<0) {num=-num;putchar('-');}
if(num>9) write(num/10);
putchar('0'+num%10);
}
void write1(ll num){write(num);putchar(' ');}
void write2(ll num){write(num);puts("");}
#define FR first
#define SE second
#define MP make_pair
#define pr pair<double,double>
#define PB push_back
#define vc vector
void chmax(ll &x,const ll y) {x=x>y?x:y;}
void chmin(ll &x,const ll y) {x=x<y?x:y;}
const int MAX_N=3e5+10;
const ll MOD=2542;
void add(ll &x,ll y) {x+=y;if(x>=MOD) x-=MOD;if(x<0) x+=MOD;}
int ff[MAX_N];vector<int> son[MAX_N];
void dfs(int x,int fa)
{
ff[x]=fa;
for(int t=0;t<(int)son[x].size();t++) if(son[x][t]!=fa) dfs(son[x][t],x);
}
int C[MAX_N];
ll tot[MAX_N];
void main()
{
int n=qread();
for(int i=1;i<n;i++){int x=qread(),y=qread();son[x].PB(y);son[y].PB(x);}dfs(1,0);
for(int i=1;i<=n;i++) C[i]=qread();
ll sum=0;for(int x=2;x<=n;x++) sum+=C[ff[x]]-C[x];tot[1]=(2*C[1]-sum)/(n-1);
for(int x=2;x<=n;x++) tot[x]=(tot[1]+C[ff[x]]-C[x])/2;
for(int x=1;x<=n;x++)
{
ll now=tot[x];
for(int t=0;t<(int)son[x].size();t++) if(son[x][t]!=ff[x]) now-=tot[son[x][t]];
write1(now);
}
}
};
int main()
{
freopen("a.in","r",stdin);
freopen("a.out","w",stdout);
srand(time(0));
mine::main();
}

本文基于 知识共享署名-相同方式共享 4.0 国际许可协议发布
本文地址:http://zory.ink/posts/2519.html
转载请注明出处,谢谢!