博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
[BJOI2018]求和
阅读量:5283 次
发布时间:2019-06-14

本文共 1737 字,大约阅读时间需要 5 分钟。

其实可以用$sum(i,j)$表示从$i$到$1$的$k$次方的值,然后就是$lca$的基本操作

注意,能一起干的事情就一起搞,要不会超时

#include
#include
#include
#include
#define int long long#define mod 998244353using namespace std;const int N=300001;inline int read(){ int f=1,ans=0;char c=getchar(); while(c<'0'||c>'9'){
if(c=='-')f=-1;c=getchar();} while(c>='0'&&c<='9'){ans=ans*10+c-'0';c=getchar();} return f*ans;}struct node{ signed u,v,nex;}x[N<<1];signed n,m,cnt;int sum[N][51];signed head[N],deep[N],fa[N][20];void add(signed u,signed v){ x[cnt].u=u,x[cnt].v=v,x[cnt].nex=head[u],head[u]=cnt++;}int ksm(int a,int b){ if(a==1) return 1; int ans=1; a%=mod; while(b){ if(b&1) ans*=a,ans%=mod; a*=a,a%=mod; b>>=1; }return ans;}void dfs(signed f,signed fath){ deep[f]=deep[fath]+1,fa[f][0]=fath; for(signed i=1;i<=50;i++) sum[f][i]=sum[fath][i]+ksm(deep[f],i); for(signed i=1;(1<
<=deep[f];++i) fa[f][i]=fa[fa[f][i-1]][i-1]; for(signed i=head[f];i!=-1;i=x[i].nex){ if(x[i].v==fath) continue; dfs(x[i].v,f); }}int Log2[N];signed lca(signed u,signed v){ if(deep[u]
=0;--i) if(deep[u]-(1<
=deep[v]) u=fa[u][i]; if(u==v) return u; for(signed i=Log2[v];i>=0;--i){ if(fa[u][i]==fa[v][i]) continue; u=fa[u][i],v=fa[v][i]; }return fa[u][0];}signed q;bool ff;signed main(){ memset(head,-1,sizeof(head)); n=read(); for(signed i=1;i
>1]+1; dfs(1,0); q=read(); while(q--){ int u=read(),v=read(),k=read(); int ls=lca(u,v); printf("%d\n",(((sum[u][k]+sum[v][k]-2*sum[ls][k]+ksm(deep[ls],k))%mod+mod)%mod)); } return 0;}
View Code

 

转载于:https://www.cnblogs.com/si-rui-yang/p/10088072.html

你可能感兴趣的文章
本机部署流程详解
查看>>
linq取DataTable某一列数据返回给一个List<int>
查看>>
SpringBoot + Spring Security 学习笔记(一)自定义基本使用及个性化登录配置
查看>>
防抖节流
查看>>
URL结构
查看>>
C#-WinForm-设计时编程【4】-响应鼠标事件
查看>>
深入Nginx
查看>>
[LeetCode] Reverse Linked List II
查看>>
ping不通的常见原因和解决办法
查看>>
AIX下绑定双网卡
查看>>
[POJ1113&POJ1696]凸包卷包裹算法和Graham扫描法应用各一例
查看>>
使用Java提供的MXBean来监控jvm创建了哪些线程
查看>>
busy monkey
查看>>
gcc 使用注意事项
查看>>
Tomcat启动时项目重复加载问题
查看>>
判断当前时间是否在一个时间段范围内
查看>>
《软件需求模式》阅读笔记02
查看>>
感知机
查看>>
php后门隐藏技巧
查看>>
SpringBoot使用Gradle构建war包
查看>>