题目描述
“特大新闻,特大新闻!全国爆发了一种极其可怕的病毒,已经开始在各个城市 中传播开来!全国陷入了巨大的危机!大量居民陷入恐慌,想要逃到其它城市以 避难!经调查显示,该病毒来自于C 市的A 学校的一次非法的……” “哎。”你关上电视,叹了口气。作为A 学校的校长,你一天前为了保命,独自 逃离了A 学校,抛弃了全校师生,包括那个曾经帮你计算并拆除道路的工程师。 你良心受到了巨大的谴责,因此决定做出一些补救,回答一些逃难的人提出的询 问。 已知该国一共有n 个城市,并且1 号城市是首都。(n-1)条双向的公路连接这些 城市,通过这些公路,任意两个城市之间存在且仅存在一条路径。每条公路有一 个长度。如果一个城市只与一条公路相连,则称它为边境城市。 该国政府有一个奇怪的规定:每个城市有一个封闭系数di,定义di为离这个城市最远的边境城市到这个城市的距离。市民们认为,一个城市的安全系数Si 和 它的封闭系数有很重要的联系。a,b,c 是该国的幸运数字,所以大家公认一个 城市的安全系数Si = (di + a) * b mod c。 市民们一共会提出m 次询问。每个询问包含三个信息,xi,yi和qi。xi是询问 者所在的城市编号。你要为这个询问者在xi 到yi 的必经之路上找出一个离xi 最近的避难城市,并且要求这个避难城市的安全系数大于等于qi。如果存在这样的城市(包含xi 和yi),则输出城市编号,否则输出一行包括一个数-1。
输入格式
第一行五个数:依次是n, m, a, b, c。 接下来n-1 行描述公路的信息。每行三个数,前两个数代表这条公路连接的两个 城市的编号,第三个数表示这条公路的长度。 再接下来m 行,每行描述一个询问,包含三个数xi, yi 和qi。
输出格式
对于每个询问,输出一行包含一个整数,存在符合要求的城市则输出城市编号, 不存在则输出-1。
输入样例
7 6 5 6 20
1 2 4
2 4 2
2 5 3
1 3 5
3 6 6
6 7 7
7 5 15
3 4 5
5 4 2
4 5 2
6 6 10
3 5 19
输出样例
6
3
2
4
6
-1
提示
【样例解释】
从城市1到城市7的安全系数依次是:18,2,8,14,0,18,0。
【数据范围】
对于100%数据, n<=100000, m<=300000, 0<xi, yi<=n; a,b,c,qi<=1,000,000, c!=0;
分析
既然每个边境城市的度数为1,那么以任意点作为根(除边境城市),边境城市一定是叶子节点。
进而推出直径的两端一定是边境城市,不然就不是最长。
先求出直径,就可以计算出每个城市的安全系数(为什么O(n)的遍历我要用Dijkstra啊啊啊啊啊啊
因为不带修改,所以考虑用倍增来维护
只需要维护最大值和最小值就好了
对于每次询问x,y,q,先找出x,y的lca,然后分别在x到lca,lca到y中找
因为到x最近,所以在x到lca中找时尽量找深度大的,在lca到y中尽量找深度小的
先考虑x到lca
对dep[x]-dep[lca]二进制拆分,如果x到x向上2^k
的祖先无解,即这中间的安全系数的最大值不满足要求,则x向上跳2^k步
如果有解,则lca向下跳到x的2^k
的祖先,但此时dep[x]-dep[lca]=2^k
,而k这一位我们已经拆分过了,所以不妨让x再向上跳一格
,使dep[x]-dep[lca]=2^k-1
就可以继续拆分了。
再考虑lca到y
此时因为要深度最小,所以要判断lca到某个节点是否有解,还要多写一个函数
同样对dep[y]-dep[lca]二进制拆分,如果y到y向上2^k
的祖先无解,则y向上跳2^k步
如果有解,因为要深度最小,还要判断y向上2^k
的祖先是否有解,如果有,则仍然是y向上跳2^k
步,如果没有,则lca向下跳到y的2^k的祖先,y向上跳一步
注意特判x,y为同一个点的情况
代码
#include<queue>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int LOG=20;
const int maxn=100005;
struct node{
int id;long long w;};
bool operator <(node a,node b){
return a.w<b.w;}
int n,m,a,b,c,p1,p2,ecnt,info[maxn],nx[maxn<<1],v[maxn<<1],w[maxn<<1];
long long d[2][maxn],dep[maxn];int val[maxn],vis[maxn],fa[maxn][25],mx[maxn][25];
void add(int u1,int v1,int w1){
nx[++ecnt]=info[u1];info[u1]=ecnt;v[ecnt]=v1;w[ecnt]=w1;}
void dfs1(int x,int f){
if(dep[x