P4178 Tree(点分治+容斥)

传送门

先点分值,求重心分治下去

然后考虑如何计算点 u u u的贡献

首先我们暴力计算出 u u u子树中到 u u u的所有距离放在数组里

排序后,双指针可以快速计算多少条路径的拼接小于等于 k k k

但是会有重复,存在两条路径都在一颗子树内部

那就容斥,对子树也计算一遍,减去子树的答案即可,减去的都是多加的

#include <bits/stdc++.h>
using namespace std;
const int maxn = 2e5+10;
struct edge
{
	int to,nxt,w;
}d[maxn]; int head[maxn],cnt=1;
void add(int u,int v,int w){ d[++cnt] = (edge){v,head[u],w},head[u] = cnt; }
int n,k,sumn,root,rem[maxn],ans,siz[maxn],mx[maxn],vis[maxn];
void getroot(int u,int fa)
{
	siz[u] = 1; mx[u] = 0;
	for(int i=head[u];i;i=d[i].nxt )
	{
		int v = d[i].to;
		if( vis[v] || v==fa )	continue;
		getroot(v,u);
		siz[u] += siz[v];
		mx[u] = max( mx[u],siz[v] );
	}
	mx[u] = max( mx[u],sumn-siz[u] );
	if( mx[u]<mx[root] )	root = u;
}
void get_dis(int u,int fa,int zhi)
{
	rem[++rem[0]] = zhi;
	for(int i=head[u];i;i=d[i].nxt )
	{
		int v = d[i].to;
		if( v==fa||vis[v] )	continue;
		get_dis(v,u,zhi+d[i].w);
	}
} 
int calc(int u,int w)//计算经过点u为根的全部答案 
{
	rem[0] = 0; get_dis(u,0,w);
	sort( rem+1,rem+1+rem[0] );
	int l = 1,r = rem[0],ans = 0;
	while( r>=l )
	{
		if( rem[l]+rem[r]<=k )	ans += r-l,l++;
		else	r--;
	}
	return ans;
}
void solve(int u)
{
	vis[u] = 1; ans += calc(u,0);
	for(int i=head[u];i;i=d[i].nxt )
	{
		int v = d[i].to;
		if( vis[v] )	continue;
		ans -= calc(v,d[i].w);
		sumn = siz[v], mx[root=0] = n;
		getroot(v,u); solve(root);
	}
}
int main()
{
	cin >> n;
	for(int i=1;i<n;i++)
	{
		int l,r,w; scanf("%d%d%d",&l,&r,&w);
		add(l,r,w); add(r,l,w);
	}
	cin >> k;
	sumn = mx[root=0] = n;
	getroot(1,0); solve(root);
	cout << ans;
}

也可以使用树状数组来统计

但是由于树状数组不能有 0 0 0,所以需要让所有权值加上一

这里偷懒,数据水,过了

#include <bits/stdc++.h>
using namespace std;
const int maxn = 5e5+10;
struct edge
{
	int to,nxt,w;
}d[maxn]; int head[maxn],cnt=1;
void add(int u,int v,int w){ d[++cnt] = (edge){v,head[u],w},head[u] = cnt; }
int n,k,sumn,root,rem[maxn],ans,siz[maxn],mx[maxn],vis[maxn],save[maxn];
struct binary_search_tree
{
		int sumn[maxn];
		int lowbit(int x){ return x&(-x); }
	public:
		void add(int x,int val){
			if( x==0 )	return;
			for( ; x<=k ;x+=lowbit(x) )	sumn[x] += val;
		}
		int ask(int x){
			if( x==0 )	return 0;
			int ans = 0;
			for( ; x ; x-=lowbit(x) )	ans += sumn[x];
			return ans;
		}
}t;
void getroot(int u,int fa)
{
	siz[u] = 1; mx[u] = 0;
	for(int i=head[u];i;i=d[i].nxt )
	{
		int v = d[i].to;
		if( vis[v] || v==fa )	continue;
		getroot(v,u);
		siz[u] += siz[v];
		mx[u] = max( mx[u],siz[v] );
	}
	mx[u] = max( mx[u],sumn-siz[u] );
	if( mx[u]<mx[root] )	root = u;
}
void get_dis(int u,int fa,int zhi)
{
	rem[++rem[0]] = zhi; save[++save[0]] = zhi;
	for(int i=head[u];i;i=d[i].nxt )
	{
		int v = d[i].to;
		if( v==fa||vis[v] )	continue;
		get_dis(v,u,zhi+d[i].w);
	}
} 
int calc(int u)//计算经过点u为根的全部答案 
{
	int ans = 0; save[0] = 0;
	for(int i=head[u];i;i=d[i].nxt )
	{
		int v = d[i].to;
		if( vis[v] )	continue;
		rem[0] = 0; get_dis(v,u,d[i].w);
		for(int j=1;j<=rem[0];j++)
		{
			if( rem[j]>k )	continue;
			int x = t.ask(k-rem[j]);
			ans += x;
		}
		for(int j=1;j<=rem[0];j++)
		{
			if( rem[j]>k )	continue;
			t.add( rem[j],1 ); ans++;
		}
	}
	for(int i=1;i<=save[0];i++)
	{
		if( save[i]>k )	continue;
		t.add( save[i],-1 );
	}
	return ans;
}
void solve(int u)
{
	vis[u] = 1; ans += calc(u);
	for(int i=head[u];i;i=d[i].nxt )
	{
		int v = d[i].to;
		if( vis[v] )	continue;
		sumn = siz[v], mx[root=0] = n;
		getroot(v,u); solve(root);
	}
}
int main()
{
	cin >> n;
	for(int i=1;i<n;i++)
	{
		int l,r,w; scanf("%d%d%d",&l,&r,&w);
		add(l,r,w); add(r,l,w);
	}
	cin >> k;
	sumn = mx[root=0] = n;
	getroot(1,0); solve(root);
	cout << ans;
}
已标记关键词 清除标记
相关推荐
【为什么还需要学习C++?】 你是否接触很多语言,但从来没有了解过编程语言的本质? 你是否想成为一名资深开发人员,想开发别人做不了的高性能程序? 你是否经常想要窥探大型企业级开发工程的思路,但苦于没有基础只能望洋兴叹? &nbsp; 那么C++就是你个人能力提升,职业之路进阶的不二之选。 【课程特色】 1.课程共19大章节,239课时内容,涵盖数据结构、函数、类、指针、标准库全部知识体系。 2.带你从知识与思想的层面从0构建C++知识框架,分析大型项目实践思路,为你打下坚实的基础。 3.李宁老师结合4大国外顶级C++著作的精华为大家推出的《征服C++11》课程。 【学完后我将达到什么水平?】 1.对C++的各个知识能够熟练配置、开发、部署; 2.吊打一切关于C++的笔试面试题; 3.面向物联网的“嵌入式”和面向大型化的“分布式”开发,掌握职业钥匙,把握行业先机。 【面向人群】 1.希望一站式快速入门的C++初学者; 2.希望快速学习 C++、掌握编程要义、修炼内功的开发者; 3.有志于挑战更高级的开发项目,成为资深开发的工程师。 【课程设计】 本课程包含3大模块 基础篇 本篇主要讲解c++的基础概念,包含数据类型、运算符等基本语法,数组、指针、字符串等基本词法,循环、函数、类等基本句法等。 进阶篇 本篇主要讲解编程中常用的一些技能,包含类的高级技术、类的继承、编译链接和命名空间等。 提升篇: 本篇可以帮助学员更加高效的进行c++开发,其中包含类型转换、文件操作、异常处理、代码重用等内容。
©️2020 CSDN 皮肤主题: Age of Ai 设计师:meimeiellie 返回首页