Jzoj3717【NOI2014模拟7.2】火车

最新推荐文章于 2021-03-03 15:01:52 发布

原创

最新推荐文章于 2021-03-03 15:01:52 发布 · 772 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#树链剖分 #树状数组

博客介绍了A国城市间独特的交通网络，其中火车必须按照特定规则行驶，即每次停在未访问过的m个城市中最靠前的一个。这个问题被识别为树剖问题，解决方案涉及使用树状数组进行区间标记。由于数据的特殊性，DFS可能导致栈溢出，因此建议从中间开始执行DFS。

A国有n个城市，城市之间有一些双向道路相连，并且城市两两之间有唯一路径。现在有火车在城市a，需要经过m个城市。火车按照以下规则行驶：每次行驶到还没有经过的城市中在m个城市中最靠前的。现在小A想知道火车经过这m个城市后所经过的道路数量。

很显然的树剖题目嘛，加上个区间打标记即可，可以用树状数组

注意数据很坑，dfs爆栈，所以要从最中间开始dfs

~~（真的不知道为什么是NOI模拟）~~

#include<stdio.h>
#include<string.h>
#include<algorithm>
#define N 500010
using namespace std;
struct Edge{ int v,nt; } G[N<<1];
int h[N],f[N],d[N],top[N],son[N],sz[N];
int n,m,cnt=0,k,w[N],v[N],tot=0; long long ans=0;
inline void adj(int x,int y){
	G[++cnt]=(Edge){y,h[x]}; h[x]=cnt;
}
inline int _(int x){ return x&-x; }
inline void add(int x,int k){ for(;x<=n;x+=_(x)) v[x]+=k; }
inline int sum(int x,int k=0){ for(;x;x-=_(x)) k+=v[x]; return k; }
void dfs(int x,int p){
	f[x]=p; d[x]=d[p]+1; sz[x]=1;
	for(int v,i=h[x];i;i=G[i].nt)
		if((v=G[i].v)!