题意:
给定一棵有 n n n 个结点的树,树结点代表一个城市,其中有些城市建立了集市,一个城市里的人会选择去最近的有集市的城市,如果有多个相同距离的,则去标号最小的。现在你可以选择在一个没有集市的城市里建立一个集市,要求最大化地被作为选择,输出有多少个城市会选择到该集市。 ( n ≤ 1 0 5 ) (n \leq 10^5) (n≤105)
链接:
https://vjudge.net/problem/HDU-5016
解题思路:
可以预处理出每个点会去的城市编号以及距离,最近距离可以通过跑最短路得到,最近的选择点可以参考最短路径树上最小字典序路径的做法,这里是多源,只需要按字典序升序从源点开始在最短路上跑
d
f
s
dfs
dfs 标记可达点。记
u
u
u 距离最近的集市距离及编号为二元组
(
a
[
u
]
,
b
[
u
]
)
(a[u], b[u])
(a[u],b[u])。
对于在每个点上建立集市的答案,点
u
u
u 选择走点
v
v
v 为一条路径,所有选择即全部路径可能,考虑点分治来统计,将所有可能的选择划分为经过根
r
t
rt
rt 的路径,记
d
i
s
[
u
]
dis[u]
dis[u] 为某次分治过程
u
u
u 到
r
t
rt
rt 的距离,则
u
u
u 选择
v
v
v 并且经过
r
t
rt
rt, 当且仅当
(
d
i
s
[
u
]
+
d
i
s
[
v
]
,
v
)
<
(
a
[
u
]
,
b
[
u
]
)
(dis[u] + dis[v], v) \lt (a[u], b[u])
(dis[u]+dis[v],v)<(a[u],b[u]),移项得
(
d
i
s
[
v
]
,
v
)
<
(
a
[
u
]
−
d
i
s
[
u
]
,
b
[
u
]
)
(dis[v], v) \lt (a[u] - dis[u], b[u])
(dis[v],v)<(a[u]−dis[u],b[u]),二元组用点分容斥写法较容易统计。
参考代码:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
#define sz(a) ((int)a.size())
#define pb push_back
#define lson (rt << 1)
#define rson (rt << 1 | 1)
#define gmid (l + r >> 1)
const int maxn = 1e5 + 5;
const int inf = 0x3f3f3f3f;
const int mod = 1e9 + 7;
vector<pii> G[maxn];
pii dis[maxn], tp[maxn];
int siz[maxn], vis[maxn], tag[maxn];
int a[maxn], b[maxn], ans[maxn];
int n, tn, rmn, rt, tot;
void dij(int dis[]){
priority_queue<pii, vector<pii>, greater<pii> > q;
for(int i = 1; i <= n; ++i) dis[i] = inf;
for(int i = 1; i <= n; ++i){
if(tag[i]) dis[i] = 0, q.push({dis[i], i});
}
while(!q.empty()){
int u = q.top().second, d = q.top().first; q.pop();
if(d != dis[u]) continue;
for(auto &e : G[u]){
int v = e.second, w = e.first;
if(dis[v] > dis[u] + w){
dis[v] = dis[u] + w;
q.push({dis[v], v});
}
}
}
}
void dfss(int u, int f){
for(auto &e : G[u]){
int v = e.second, w = e.first;
if(v == f || b[v] || a[v] != a[u] + w) continue;
b[v] = b[u];
dfss(v, u);
}
}
void getRt(int u, int f){
int mx = 0; siz[u] = 1;
for(auto &e : G[u]){
int v = e.second;
if(v == f || vis[v]) continue;
getRt(v, u);
siz[u] += siz[v];
mx = max(mx, siz[v]);
}
mx = max(mx, tn - siz[u]);
if(mx < rmn) rmn = mx, rt = u;
}
void dfs(int u, int f, int d){
if(tag[u]) return;
dis[++tot] = {d, u};
for(auto &e : G[u]){
int v = e.second, w = e.first;
if(v == f || vis[v]) continue;
dfs(v, u, d + w);
}
}
void cal(int u, int d, int flg){
tot = 0;
dfs(u, 0, d);
for(int i = 1; i <= tot; ++i){
int v = dis[i].second, d = dis[i].first;
tp[i] = {a[v] - d, b[v]};
}
sort(tp + 1, tp + 1 + tot);
for(int i = 1; i <= tot; ++i){
int p = upper_bound(tp + 1, tp + 1 + tot, dis[i]) - tp;
ans[dis[i].second] += flg * (tot - p + 1);
}
}
void dfz(int u){
vis[u] = 1;
cal(u, 0, 1);
for(auto &e : G[u]){
int v = e.second, w = e.first;
if(vis[v]) continue;
cal(v, w, -1);
tn = siz[v], rmn = inf, getRt(v, u);
dfz(rt);
}
vis[u] = 0;
}
int main(){
ios::sync_with_stdio(0); cin.tie(0);
while(cin >> n){
for(int i = 1; i <= n; ++i){
G[i].clear();
ans[i] = 0;
b[i] = 0;
}
for(int i = 1; i < n; ++i){
int u, v, w; cin >> u >> v >> w;
G[u].pb({w, v}), G[v].pb({w, u});
}
for(int i = 1; i <= n; ++i) cin >> tag[i];
dij(a);
for(int i = 1; i <= n; ++i){
if(tag[i]) b[i] = i, dfss(i, 0);
}
tn = n, rmn = inf, getRt(1, 0);
dfz(rt);
int ret = 0;
for(int i = 1; i <= n; ++i){
// cout << i << " ? " << a[i] << " " << b[i] << " " << ans[i] << endl;
ret = max(ret, ans[i]);
}
cout << ret << "\n";
}
return 0;
}