[Luogu3806] 点分治模板1 [点分治]

最新推荐文章于 2022-07-08 16:28:10 发布
最新推荐文章于 2022-07-08 16:28:10 发布
阅读量221 收藏
点赞数
文章标签： 点分治
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/Estia_/article/details/82902255
版权

题意：给定一棵有 n n n个点的树和 m m m个询问，每个询问要求回答树上距离为 k k k的点对是否存在。
n &lt; = 1 0 4 , m &lt; = 1 0 2 , w i &lt; = 1 0 3 , k &lt; = 1 0 7 n&lt;=10^4,m&lt;=10^2,w_i&lt;=10^3,k&lt;=10^7 n<=104,m<=102,wi​<=103,k<=107

树上+距离+点对≈点分治（

一开始简单考虑得到这么一个写法

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstdlib>
#include<cctype>
#include<cmath>
#include<cstring>
using namespace std;
#define getchar() (frS==frT&&(frT=(frS=frBB)+fread(frBB,1,1<<12,stdin),frS==frT)?EOF:*frS++)
char frBB[1<<12]={},*frS=frBB,*frT=frBB;
inline int read()
{
	int x=0;char ch=getchar();
	while(!isdigit(ch))ch=getchar();
	while(isdigit(ch))x=x*10+(ch-'0'),ch=getchar();
	return x;
}
#define add_edge(a,b,c) nxt[++tot]=head[a],head[a]=tot,to[tot]=b,val[tot]=c
int N,M,tot=0,mn=2147483647,curtotal=0,root=0;
int head[10005]={},to[20005]={},val[20005]={},nxt[20005]={};
int k[105]={};
int siz[10005]={},mxsiz[10005]={};
bool vis[10005]={};
int ans[105]={};
int dis[10005]={};
void getroot(int x,int fa)
{
	mxsiz[x]=0,siz[x]=1;
	for(int i=head[x];i;i=nxt[i])
	{
		if(to[i]==fa||vis[to[i]])continue;
		getroot(to[i],x);
		mxsiz[x]=max(mxsiz[x],siz[to[i]]);
		siz[x]+=siz[to[i]];
	}
	mxsiz[x]=max(mxsiz[x],curtotal-siz[x]);
	if(mxsiz[x]<mn)mn=mxsiz[x],root=x;
}
int getdis(int x,int fa,int l)
{
	dis[++dis[0]]=l;
	for(int i=head[x];i;i=nxt[i])
	{
		if(vis[to[i]]||to[i]==fa)continue;
		getdis(to[i],x,l+val[i]);
	}
}
void solve(int x,int add,int opt)
{
	dis[0]=0; getdis(x,0,add);
	sort(dis+1,dis+1+dis[0]);
	for(int i=1;i<=M;++i)
	for(int r=dis[0],l=1;l<dis[0];++l)
	{
		while((dis[r]+dis[l]>k[i])&&(r>l))--r;
		if(r<=l)break;
		if(dis[r]+dis[l]==k[i])ans[i]+=opt;
	}
}
void divide(int x)
{
	solve(x,0,1); vis[x]=1;
	for(int i=head[x];i;i=nxt[i])
	{
		if(vis[to[i]])continue;
		solve(to[i],val[i],-1);
		mn=2147483647,curtotal=0,getroot(to[i],0);
		divide(root);
	}
}
int main()
{
	freopen("fa.in","r",stdin); freopen("fa.out","w",stdout);
	N=read();M=read();
	for(int a,b,c,i=1;i<N;++i)
	{
		a=read();b=read();c=read();
		add_edge(a,b,c); add_edge(b,a,c);
	}
	for(int i=1;i<=M;++i)k[i]=read();
	mn=2147483647,curtotal=0,getroot(1,0);
	divide(root);
	for(int i=1;i<=M;++i)printf((ans[i])?"AYE\n":"NAY\n");
	return 0;
}

然后T惨了 复杂度好像是 Θ ( 100 N l o g 2 N ) \Theta(100Nlog^2N) Θ(100Nlog2N)？

稍微修改了一下。↓

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstdlib>
#include<cctype>
#include<cmath>
#include<cstring>
using namespace std;
#define getchar() (frS==frT&&(frT=(frS=frBB)+fread(frBB,1,1<<12,stdin),frS==frT)?EOF:*frS++)
char frBB[1<<12]={},*frS=frBB,*frT=frBB;
inline int read()
{
    int x=0;
	register char c=getchar();
    for(;!isdigit(c);c=getchar());
    for(;isdigit(c);x=x*10+c-'0',c=getchar());
    return x;
}
#define add_edge(a,b,c) nxt[++tot]=head[a],head[a]=tot,to[tot]=b,val[tot]=c
int N,M,tot=0,mn=2147483647,curtotal,root=0;
int head[10005]={},to[20005]={},val[20005]={},nxt[20005]={};
int k[10005]={};
int siz[10005]={},mxsiz[10005]={};
bool vis[10005]={};
bool ans[10005]={};
int dis[10005]={};
bool exist[10000005]={};
int ms[10005]={};
void getroot(int x,int fa)
{
	mxsiz[x]=0,siz[x]=1;
	for(int i=head[x];i;i=nxt[i])
	{
		if((!vis[to[i]])&&(to[i]!=fa))
		{
			getroot(to[i],x);
			siz[x]+=siz[to[i]];
			if(siz[to[i]]>mxsiz[x])mxsiz[x]=siz[to[i]];
		}
	}
	mxsiz[x]=max(mxsiz[x],curtotal-siz[x]);
	if(mxsiz[x]<mn)mn=mxsiz[x],root=x;
}
int getdis(int x,int fa,int l)
{
	dis[++dis[0]]=l;
	for(int i=head[x];i;i=nxt[i])
		if((!vis[to[i]])&&(to[i]!=fa))getdis(to[i],x,l+val[i]);
}
void solve(int x,int add)
{
	dis[0]=0; getdis(x,0,add);
	for(int i=1;i<=dis[0];++i)
		for(int j=1;j<=M;++j)
			if(k[j]>=dis[i])ans[j]|=exist[k[j]-dis[i]];
			
	for(int i=1;i<=dis[0];++i)exist[dis[i]]=1,ms[++ms[0]]=dis[i];
}
void divide(int x)
{
	vis[x]=exist[0]=1; ms[0]=0;
	for(int i=head[x];i;i=nxt[i])if(!vis[to[i]])solve(to[i],val[i]);
	for(int i=1;i<=ms[0];++i)exist[ms[i]]=0;
	for(int i=head[x];i;i=nxt[i])
	{
		if(!vis[to[i]])
		{
			mn=N,curtotal=siz[to[i]],getroot(to[i],x);
			divide(root);
		}
	}
}
int main()
{
	N=read();M=read();
	for(int a,b,c,i=1;i<N;++i)
	{
		a=read();b=read();c=read();
		add_edge(a,b,c); add_edge(b,a,c);
	}
	for(int i=1;i<=M;++i)k[i]=read();
	mn=N,curtotal=N,getroot(1,0);
	divide(root);
	for(int i=1;i<=M;++i)puts((ans[i])?"AYE":"NAY");
	return 0;
}

本来写完上面这份代码以为能A了，结果还是T
我就想咋回事 然后发现curtotal设错了 改好就A了
然后发现一开始写的代码改好curtotal也A了↓

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstdlib>
#include<cctype>
#include<cmath>
#include<cstring>
using namespace std;
#define getchar() (frS==frT&&(frT=(frS=frBB)+fread(frBB,1,1<<12,stdin),frS==frT)?EOF:*frS++)
char frBB[1<<12]={},*frS=frBB,*frT=frBB;
inline int read()
{
    int x=0;char ch=getchar();
    while(!isdigit(ch))ch=getchar();
    while(isdigit(ch))x=x*10+(ch-'0'),ch=getchar();
    return x;
}
#define add_edge(a,b,c) nxt[++tot]=head[a],head[a]=tot,to[tot]=b,val[tot]=c
int N,M,tot=0,mn=2147483647,curtotal=0,root=0;
int head[10005]={},to[20005]={},val[20005]={},nxt[20005]={};
int k[105]={};
int siz[10005]={},mxsiz[10005]={};
bool vis[10005]={};
int ans[105]={};
int dis[10005]={};
void getroot(int x,int fa)
{
    mxsiz[x]=0,siz[x]=1;
    for(int i=head[x];i;i=nxt[i])
    {
        if(to[i]==fa||vis[to[i]])continue;
        getroot(to[i],x);
        mxsiz[x]=max(mxsiz[x],siz[to[i]]);
        siz[x]+=siz[to[i]];
    }
    mxsiz[x]=max(mxsiz[x],curtotal-siz[x]);
    if(mxsiz[x]<mn)mn=mxsiz[x],root=x;
}
int getdis(int x,int fa,int l)
{
    dis[++dis[0]]=l;
    for(int i=head[x];i;i=nxt[i])
    {
        if(vis[to[i]]||to[i]==fa)continue;
        getdis(to[i],x,l+val[i]);
    }
}
void solve(int x,int add,int opt)
{
    dis[0]=0; getdis(x,0,add);
    sort(dis+1,dis+1+dis[0]);
    for(int i=1;i<=M;++i)
    for(int r=dis[0],l=1;l<dis[0];++l)
    {
        while((dis[r]+dis[l]>k[i])&&(r>l))--r;
        if(r<=l)break;
        if(dis[r]+dis[l]==k[i])ans[i]+=opt;
    }
}
void divide(int x)
{
    solve(x,0,1); vis[x]=1;
    for(int i=head[x];i;i=nxt[i])
    {
        if(vis[to[i]])continue;
        solve(to[i],val[i],-1);
        mn=N,curtotal=siz[to[i]],getroot(to[i],0);
        divide(root);
    }
}
int main()
{
    N=read();M=read();
    for(int a,b,c,i=1;i<N;++i)
    {
        a=read();b=read();c=read();
        add_edge(a,b,c); add_edge(b,a,c);
    }
    for(int i=1;i<=M;++i)k[i]=read();
    mn=N,curtotal=N,getroot(1,0);
    divide(root);
    for(int i=1;i<=M;++i)printf((ans[i])?"AYE\n":"NAY\n");
    return 0;
}

怎么讲呢 感觉微妙

好像复杂度其实是 Θ ( K N l o g N ) \Theta(KNlogN) Θ(KNlogN)？

_er
关注
  • 0
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
vscode-luogu:解决VSCode中落谷问题
05-30
vscode-luogu轻松的让你在VSCode 上提交洛谷代码使用说明查看题目按下F1或Ctrl+Shift+P ，查找Luogu: 查看题目命令，点击后输入题号完成搜索点击界面左侧洛谷图片，然后点击搜索图标，输入题号完成搜索完成功能查看...
洛谷P3806 点分治模板
Chase的博客
07-19 180
题意： 点分治模板，问树上两点路径权值和为kkk的无序对数目。 分析： 点分治：每次在无根树中选取一个点为根，然后递归处理以根节点的儿子为根的子树。(选取哪个为根会影响效率) 人话：递归子树 对于以rootrootroot为根的树，那么树上两点u,vu,vu,v路径有两种情况： 1.1.1.路径经过rootrootroot 2.2.2.路径不经过rootrootroot 我们可以只统计路径经过ro...
luoguP3806 【模板】点分治1 [点分治]
dirushi6322的博客
08-13 96
题目背景 感谢hzwer的点分治互测。 题目描述 给定一棵有n个点的树 询问树上距离为k的点对是否存在。 输入输出格式 输入格式： n,m 接下来n-1条边a,b,c描述a到b有一条长度为c的路径 接下来m行每行询问一个K 输出格式： 对于每个K每行输出一个答案，存在输出“AYE”,否则输出”NAY”(不包含引号) 输入输出样例 输入样例#1： 2 1 ...
点分治模板
b10221001的博客
08-07 69
点分治模板 点分治： 这里统一地，只讲一下主要思想，各种细节留给参考文章。 点分治的主要是处理树上的路径问题，如果用暴力做法求解会有很高的时间复杂度\(O(n^2)-O(n^3)\)的样子。而如果使用点分治，就可以将复杂度降为\(O(nlog_2n)\)。主要思路是将一棵树分解成若干子树，在分解成子树的子树，递归地求解出部分答案（分），最后在递归回去时合并答案（治）。至于怎么分解树，这里...
点分治模板 / p3806
m0_54646258的博客
07-08 194
点分治是一种解决树上点对、树上路径的问题的分治方法，有着优秀的时间复杂度，可以处理大规模的树上问题
luogu3806 点分治模板
pubgoso的博客
09-16 147
题目链接 每次找重心，然后处理每个子树。 // luogu-judger-enable-o2 #include<bits/stdc++.h> #define fi first #define se second #define pb push_back #define mp make_pair #define LL long long #define SZ(X) X.size() #...
luogu8月比赛PPT讲解
08-10
这个是luogu的PPT 8月的比赛 讲解PPT 挺好的 没有看过的可以去luogu看看，题目挺好的；
luogu-cli：罗古在线裁判的cli工具
03-04
如果luogu-cli在运行时产生了类似下面的错误： /usr/bin/env: “node \r ”: 没有那个文件或目录 请尝试下面的操作： 罗古里 洛谷命令行工具 支持以下功能： 译文查看 译文提交 译文搜索 奔奔发送 奔奔查看 云剪切...
洛谷 [模板]KMP字符串匹配
01-03
题目链接：https://www.luogu.com.cn/problem/P3375 #include using namespace std; int Next[1000010]={0}; void getNEXT(string str) { Next[0] = -1; int i=-1, j=0; while(j < str.size()) { if(i==-1 |...
后缀自动机模板
01-07
入门推荐博客：https://www.luogu.com.cn/blog/Kesdiael3/hou-zhui-zi-dong-ji-yang-xie  简单介绍一下各个参数：构造SAM时强制在线...到达点 i 的所有字符串的长度都必然大于到达 fa(i) 的所有字符串的长度，且到
洛谷 3806【模板】点分治1——点分治
weixin_34358365的博客
08-14 84
题目：https://www.luogu.org/problemnew/show/P3806 点分治的第二道题！ 和点分治的第一道题一模一样！就是多了一些询问罢了。（然而自己却写得漏洞百出，WA了数次……） 注意这回有边权为0的边，所以在 f [ sta[ i ] ] = 0 的时候可能把 f [ 0 ] 也赋成了0，要每次赋一下1。 #include<iostream> ...
洛谷 P3806 【模板】点分治1 点分治模版题
liangzihao1的博客
03-07 355
题目描述给定一棵有n个点的树询问树上距离为k的点对是否存在。输入输出格式输入格式： n,m 接下来n-1条边a,b,c描述a到b有一条长度为c的路径接下来m行每行询问一个K输出格式： 对于每个K每行输出一个答案，存在输出“AYE”,否则输出”NAY”(不包含引号)输入输出样例输入样例#1： 2 1 1 2 2 2 输出样例#1： AYE 说明对于30%的数据n<=100对于60%
点分治[模板][luogu3806]
FSYo 的博客
10-29 137
传送门 我们直接把可以的dis存在数组里,注意solve儿子时先剪掉贡献,具体看代码吧 #include&lt;bits/stdc++.h&gt; #define N 40005 using namespace std; int first[N],next[N],to[N],w[N],tot; int n,m,siz,vis[N],ret,K[N]; int size[N],rt,Maxso...
点分治-luogu3806
weixin_44203780的博客
05-28 125
题目链接 luogu3806 思路 这题是点分治的模板题目，点分治就是一棵树上的分治。 详细的教程上B站看AgOH的直播，很感谢大佬的分享 点分治教程-作者:AgOH 代码实现 #include<bits/stdc++.h> using namespace std; #define ll long long const int N = 1e4+10; const int NN = 1e7+10; struct edge{ int v,dis,nex; }edges[N<<1]
洛谷-3806 【模板】点分治1
mkopvec的博客
08-03 125
题目描述 给定一棵有n个点的树 询问树上距离为k的点对是否存在。 输入格式 n,m 接下来n-1条边a,b,c描述a到b有一条长度为c的路径 接下来m行每行询问一个K 输出格式 对于每个K每行输出一个答案，存在输出“AYE”,否则输出”NAY”(不包含引号) 输入输出样例 输入 #1 复制 2 1 1 2 2 2 输出 #1 复制 AYE 说明/提示 对于30%的数据n<=10...
洛谷P3806 【模板】点分治1(点分治 模板题)
Code92007的博客
10-10 338
题目 给定一棵有n(n<=1e4)个点的树 询问树上距离为k(k<=1e7)的点对是否存在 询问m(m<=100)组 思路来源 https://www.cnblogs.com/bztMinamoto/p/9489473.html https://baijiahao.baidu.com/s?id=1608747083240620752&wfr=spider&am...
luogu1P1923 python
最新发布
08-14
根据提供的引用内容，我无法找到与"luogu1P1923"相关的信息。请提供更多的上下文或问题细节，以便我能够更好地回答你的问题。<span class="em">1</span><span class="em">2</span><span class="em">3 #### 引用[....

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值