331 树分治 点分治
2022/8/23 23:25:42
本文主要是介绍331 树分治 点分治,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!
视频链接:
Luogu P3806 【模板】点分治1
#include<iostream> #include<algorithm> using namespace std; const int N=10005; const int INF=10000005; struct node{int v,w,ne;}e[N<<1]; int idx,h[N]; //加边 int del[N],siz[N],mxs,sum,root;//求根 int dis[N],d[N],cnt; //求距离 int ans[N],q[INF],judge[INF];//判答案 int n,m,ask[N]; void add(int u,int v,int w){ e[++idx].v=v; e[idx].w=w; e[idx].ne=h[u]; h[u]=idx; } void getroot(int u,int fa){ siz[u]=1; int s=0; for(int i=h[u];i;i=e[i].ne){ int v=e[i].v; if(v==fa||del[v])continue; getroot(v,u); siz[u]+=siz[v]; s=max(s,siz[v]); } s=max(s,sum-siz[u]); if(s<mxs) mxs=s, root=u; } void getdis(int u,int fa){ dis[++cnt]=d[u]; for(int i=h[u];i;i=e[i].ne){ int v=e[i].v; if(v==fa||del[v])continue; d[v]=d[u]+e[i].w; getdis(v,u); } } void divide(int u){ del[u]=judge[0]=1; int p=0; // 枚举u的子树,判定答案 for(int i=h[u];i;i=e[i].ne){ int v=e[i].v; if(del[v])continue; //求出子树v的各点到u的距离 cnt=0; d[v]=e[i].w; getdis(v,u); // 枚举距离和询问,判定答案 for(int j=1;j<=cnt;++j) for(int k=1;k<=m;++k) if(ask[k]>=dis[j]) ans[k]|=judge[ask[k]-dis[j]]; // 记录合法距离 for(int j=1;j<=cnt;++j) if(dis[j]<INF) q[++p]=dis[j], judge[dis[j]]=1; } for(int i=1;i<=p;++i)judge[q[i]]=0; // 枚举u的子树,求根分治 for(int i=h[u];i;i=e[i].ne){ int v=e[i].v; if(del[v])continue; mxs=sum=siz[v]; getroot(v,0); //求根 divide(root); //分治 } } int main(){ scanf("%d%d",&n,&m); for(int i=1;i<n;++i){ int u,v,w; scanf("%d%d%d",&u,&v,&w); add(u,v,w);add(v,u,w); } for(int i=1;i<=m;++i) scanf("%d",&ask[i]); mxs=sum=n; getroot(1,0); getroot(root,0); //重构siz[] divide(root); for(int i=1;i<=m;++i) ans[i]?puts("AYE"):puts("NAY"); return 0; }
这篇关于331 树分治 点分治的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!
- 2024-05-01为什么公共事业机构会偏爱 TiDB :TiDB 数据库在某省妇幼健康管理系统的应用
- 2024-04-26敏捷开发:想要快速交付就必须舍弃产品质量?
- 2024-04-26静态代码分析的这些好处,我竟然都不知道?
- 2024-04-26你在测试金字塔的哪一层?(下)
- 2024-04-26快刀斩乱麻,DevOps让代码评审也自动起来
- 2024-04-262024年最好用的10款ER图神器!
- 2024-04-2203-为啥大模型LLM还没能完全替代你?
- 2024-04-2101-大语言模型发展
- 2024-04-17基于SpringWeb MultipartFile文件上传、下载功能
- 2024-04-14个人开发者,Spring Boot 项目如何部署