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-=_