BZOJ3697: 采药人的路径(点分治)
2024-08-31 17:45:39
Description
采药人的药田是一个树状结构,每条路径上都种植着同种药材。
采药人以自己对药材独到的见解,对每种药材进行了分类。大致分为两类,一种是阴性的,一种是阳性的。
采药人每天都要进行采药活动。他选择的路径是很有讲究的,他认为阴阳平衡是很重要的,所以他走的一定是两种药材数目相等的路径。采药工作是很辛苦的,所以他希望他选出的路径中有一个可以作为休息站的节点(不包括起点和终点),满足起点到休息站和休息站到终点的路径也是阴阳平衡的。他想知道他一共可以选择多少种不同的路径。
Input
第1行包含一个整数N。
接下来N-1行,每行包含三个整数a_i、b_i和t_i,表示这条路上药材的类型。
Output
输出符合采药人要求的路径数目。
Sample Input
7
1 2 0
3 1 1
2 4 0
5 2 0
6 3 1
5 7 1
1 2 0
3 1 1
2 4 0
5 2 0
6 3 1
5 7 1
Sample Output
1
解题思路:
点分治要求在寻找到一条链时统计答案与重心在链的哪个位置无关。而显然这道题如果枚举重心为中转站是错误的,因为一条链只被统计一次,而中心位置很可能是错误的,所以我们需要修正这一点,就是统计重心路径上可能出现的中心位置。
换句话说,开四个桶来记录,分别记录当前子树内可能出现的中转站(出现过的距离重复,在差分的意义下说明出现和为0的区间),另外一个就是之前子树的答案,每次更新完合并桶就好了。
代码:
#include<cstdio>
#include<cstring>
#include<algorithm>
typedef long long lnt;
const int N=;
struct pnt{
int hd;
int dis;
int wgt;
bool vis;
}p[N];
struct ent{
int twd;
int lst;
int vls;
}e[N<<];
lnt f[N<<][],g[N<<][];
int has[N<<];
int n;
lnt top;
int cnt;
lnt ans;
int lim;
int size;
int root;
int maxsize;
void ade(int f,int t,int v)
{
cnt++;
e[cnt].twd=t;
e[cnt].lst=p[f].hd;
e[cnt].vls=v;
p[f].hd=cnt;
return ;
}
void grc_dfs(int x,int f)
{
p[x].wgt=;
int maxs=-;
for(int i=p[x].hd;i;i=e[i].lst)
{
int to=e[i].twd;
if(to==f||p[to].vis)
continue;
grc_dfs(to,x);
p[x].wgt+=p[to].wgt;
if(p[to].wgt>maxs)
maxs=p[to].wgt;
}
if(maxs<size-p[x].wgt)
maxs=size-p[x].wgt;
if(maxs<maxsize)
{
maxsize=maxs;
root=x;
}
return ;
}
void ans_dfs(int x,int fa,int dep)
{
lim=std::max(lim,dep);
if(has[p[x].dis])
f[p[x].dis][]++;
else
f[p[x].dis][]++;
has[p[x].dis]++;
for(int i=p[x].hd;i;i=e[i].lst)
{
int to=e[i].twd;
if(to==fa||p[to].vis)
continue;
p[to].dis=p[x].dis+e[i].vls;
ans_dfs(to,x,dep+);
}
has[p[x].dis]--;
return ;
}
void bin_dfs(int x)
{
int maxd=;
p[x].vis=true;
g[n][]=;
for(int i=p[x].hd;i;i=e[i].lst)
{
int to=e[i].twd;
if(p[to].vis)
continue;
p[to].dis=n+e[i].vls;
lim=;
ans_dfs(to,to,);
maxd=std::max(maxd,lim);
ans+=f[n][]*(g[n][]-);
for(int j=-lim;j<=lim;j++)
ans+=g[n-j][]*f[n+j][]+g[n-j][]*f[n+j][]+g[n-j][]*f[n+j][];
for(int j=n-lim;j<=n+lim;j++)
{
g[j][]+=f[j][];
g[j][]+=f[j][];
f[j][]=f[j][]=;
}
}
for(int i=n-maxd;i<=n+maxd;i++)
g[i][]=g[i][]=;
for(int i=p[x].hd;i;i=e[i].lst)
{
int to=e[i].twd;
if(p[to].vis)
continue;
root=;
size=p[to].wgt;
maxsize=0x3f3f3f3f;
grc_dfs(to,to);
bin_dfs(root);
}
return ;
}
int main()
{
scanf("%d",&n);
for(int i=;i<n;i++)
{
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
c=c*-;
ade(a,b,c);
ade(b,a,c);
}
root=;
size=n;
maxsize=0x3f3f3f3f;
grc_dfs(,);
bin_dfs(root);
printf("%lld\n",ans);
return ;
}
最新文章
- Cocos2d-x 3.0 Json用法 Cocos2d-x xml解析
- nodejs express 框架解密3-中间件模块
- JavaBean 动作元素事例
- struts几个配置文件加载顺序_2015.01.04
- Android -- FragmentTabHost实现微信底部切换
- DF学Mysql(二)——数据表的基本操作
- VMware系统运维(十五)部署虚拟化桌面Horizon View Manager 5.2添加vCenter Server服务器
- OSA-MAC: A MAC Protocol for Opportunistic Spectrum Access in Cognitive Radio Networks
- MYSQL 的 6 个返回时间日期函数
- Scrapy的架构初探
- supervisor 安装配置
- Visual Studio 2015中设计UML类图
- 18.1 volatile的作用
- Mac android studio 一直卡在Gradle:Build Running的解决办法
- PHP发起POST DELETE GET POST 请求
- Java考试题之八
- [NPM] Execute Code from a Remote GitHub Branch with npx
- JSTL 标签库<;转>;
- x-pack 功能介绍及配置传输层安全性(TLS / SSL)
- JS获取当前时间及时间戳相互转换
热门文章
- jquery 表单重置通用方法
- Hyper-v Server安装与配置-新加GUI界面配置工具介绍
- C#一些延时函数
- 客户端本地存储(cookie、web Storage、vuex)选择
- 测试理论--web测试方法总结
- 发送邮件被退回,提示: Helo command rejected: Invalid name 错误
- 【HDOJ 5384】Danganronpa
- [转]Massive Model Rendering Techniques
- es67
- windows 常见环境变量(%AppData%、%TEMP%、%TMP%)