社区讨论
这个淀粉质她只有60分 QAQ
P4178Tree参与者 7已保存回复 12
讨论操作
快速查看讨论及其快照的属性,并进行相关操作。
- 当前回复
- 12 条
- 当前快照
- 1 份
- 快照标识符
- @mi7ppwg3
- 此快照首次捕获于
- 2025/11/21 01:35 4 个月前
- 此快照最后确认于
- 2025/11/21 01:53 4 个月前
T飞了
CPP#pragma GCC optimize (3)
#include<cstdio>
#include<cmath>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<vector>
#include<queue>
#include<bitset>
#define N 40002
#define inf 0x3f3f3f3f
using namespace std;
struct edge{
int v,w;
edge(int v=0,int w=0):v(v),w(w){}
};
vector<edge> adj[N];
int dis[N],size[N],son[N],node[N];
bool vis[N];
int n,m,r,k,sum,cnt,ans;
inline void read(int &x);
void get_root(int u,int fa);
void partition(int u);
void get_dist(int u,int fa,int len);
void solve(int u,int len,int w);
int main(){
int u,v,w;
read(n);
for(int i=1;i<n;++i){
read(u),read(v),read(w);
adj[u].push_back(edge(v,w));
adj[v].push_back(edge(u,w));
}
read(k);
sum = son[r] = n;
get_root(1,0);
partition(r);
printf("%d",ans);
return 0;
}
void solve(int u,int len,int w){
cnt = 0;
dis[u] = len;
get_dist(u,0,len);
for(int i=1;i<cnt;++i){
for(int j=i+1;j<=cnt;++j){
if(node[i]+node[j]>k) continue;
ans += w;
}
}
}
void get_dist(int u,int fa,int len){
int v,l = adj[u].size();
node[++cnt] = dis[u];
for(int i=0;i<l;++i){
v = adj[u][i].v;
if(v==fa||vis[v]) continue;
dis[v] = len+adj[u][i].w;
get_dist(v,u,dis[v]);
}
}
void partition(int u){
solve(u,0,1);
vis[u] = true;
int v,l = adj[u].size();
for(int i=0;i<l;++i){
v = adj[u][i].v;
if(vis[v]) continue;
solve(v,adj[u][i].w,-1);
sum = size[v];
r = 0;
son[r] = n;
get_root(v,u);
partition(r);
}
}
void get_root(int u,int fa){
size[u] = 1,son[u] = 0;
int v,l = adj[u].size();
for(int i=0;i<l;++i){
v = adj[u][i].v;
if(v==fa||vis[v]) continue;
get_root(v,u);
size[u] += size[v];
son[u] = max(son[u],size[v]);
}
son[u] = max(son[u],sum-size[u]);
if(son[u]<son[r]) r = u;
}
inline void read(int &x){
x = 0;
char c = getchar();
while(c<'0'||c>'9') c = getchar();
while(c>='0'&&c<='9'){
x = (x<<3)+(x<<1)+(c^48);
c = getchar();
}
}
回复
共 12 条回复,欢迎继续交流。
正在加载回复...