Luogu P4427 [BJOI2018]求和
这是一道巨狗题,我已无力吐槽为什么我怎么写都不过
我们对于这种无修改的边权题目有一个经典的树上差分套路:
\(ans=sum_x+sum_y-2\cdot sum_{LCA(x,y)}\)
这里的\(sum\)表示的是从根到这个点的边权前缀和
然后这里求的是点权,我们还是用一样的策略,就是树上查分后加上这个点的点权即可,即:
\(ans=sum_x+sum_y-2\cdot sum_{LCA(x,y)}+node_{LCA(x,y)}\)
然后我们发现\(1\le k\le 50\),所以我们预处理出所有的幂次的情况然后LCA查找即可。
然后就我就先写了我比较熟悉的DFS序+RMQ,然后蜜汁RE90pts
DFS序+RMQ CODE
// luogu-judger-enable-o2
#include<cstdio>
#include<cstring>
#include<cmath>
using namespace std;
typedef long long LL;
const LL N=300005,P=25,mod=998244353;
struct edge
{
LL to,next;
}e[N<<1];
struct RMQ
{
LL x,num;
}f[N<<1][P];
LL head[N],dep[N],fir[N],n,m,k,cnt,tot,x,y,rt=1,s[N][55],node[N][55];
inline char tc(void)
{
static char fl[100000],*A=fl,*B=fl;
return A==B&&(B=(A=fl)+fread(fl,1,100000,stdin),A==B)?EOF:*A++;
}
inline void read(LL &x)
{
x=0; char ch=tc();
while (ch<'0'||ch>'9') ch=tc();
while (ch>='0'&&ch<='9') x=x*10+ch-'0',ch=tc();
}
inline void write(LL x)
{
if (x/10) write(x/10);
putchar(x%10+'0');
}
inline void add(LL x,LL y)
{
e[++cnt].to=y; e[cnt].next=head[x]; head[x]=cnt;
}
inline void swap(LL &a,LL &b)
{
LL t=a; a=b; b=t;
}
inline void DFS(LL now,LL fa,LL d)
{
dep[now]=d; f[++tot][0].x=dep[now]; f[tot][0].num=now; fir[now]=tot;
for (register LL i=head[now];i^-1;i=e[i].next)
if (e[i].to^fa) DFS(e[i].to,now,d+1),f[++tot][0].x=dep[now],f[tot][0].num=now;
}
inline void reset(LL now,LL fa,LL k)
{
node[now][k]=(node[now][k-1]*dep[now])%mod; s[now][k]=(s[now][k]+node[now][k])%mod;;
for (register LL i=head[now];i^-1;i=e[i].next)
if (e[i].to^fa) s[e[i].to][k]=(s[e[i].to][k]+s[now][k])%mod,reset(e[i].to,now,k);
}
inline void init(void)
{
for (register LL j=1;j<P;++j)
for (register LL i=1;i+(1<<j)-1<=tot;++i)
f[i][j]=f[i][j-1].x<f[i+(1<<j-1)][j-1].x?f[i][j-1]:f[i+(1<<j-1)][j-1];
}
inline LL LCA(LL x,LL y)
{
x=fir[x]; y=fir[y]; if (x>y) swap(x,y);
LL k=(LL)log2(y-x+1);
return f[x][k].x<f[y-(1<<k)+1][k].x?f[x][k].num:f[y-(1<<k)+1][k].num;
}
int main()
{
//freopen("CODE.in","r",stdin); freopen("CODE.out","w",stdout);
register LL i; read(n);
memset(head,-1,sizeof(head));
memset(e,-1,sizeof(e));
for (i=1;i<n;++i)
read(x),read(y),add(x,y),add(y,x);
DFS(rt,-1,0); init();
for (i=1;i<=n;++i)
node[i][0]=1;
for (i=1;i<=50;++i)
reset(rt,-1,i); read(m);
while (m--)
{
read(x); read(y); read(k); LL fa=LCA(x,y);
write((LL)((s[x][k]+s[y][k]-(2*s[fa][k])%mod+node[fa][k])%mod+mod)%mod); putchar('\n');
}
return 0;
}
然后我就不爽了,直接上了大力树剖,当然很不幸,树剖被Hack数据卡掉了
大力树剖CODE
// luogu-judger-enable-o2
#include<cstdio>
#include<cstring>
using namespace std;
typedef long long LL;
const int N=300005,K=55,mod=998244353;
struct edge
{
int to,next;
}e[N<<1];
int n,m,rt=1,x,y,k,cnt,tot,head[N],dep[N][K],tree[N<<2][K],father[N],id[N],top[N],son[N],size[N],num[N];
inline char tc(void)
{
static char fl[100000],*A=fl,*B=fl;
return A==B&&(B=(A=fl)+fread(fl,1,100000,stdin),A==B)?EOF:*A++;
}
inline void read(int &x)
{
x=0; char ch=tc();
while (ch<'0'||ch>'9') ch=tc();
while (ch>='0'&&ch<='9') x=x*10+ch-'0',ch=tc();
}
inline void write(int x)
{
if (x/10) write(x/10);
putchar(x%10+'0');
}
inline void add(int x,int y)
{
e[++cnt].to=y; e[cnt].next=head[x]; head[x]=cnt;
}
inline void swap(int &a,int &b)
{
int t=a; a=b; b=t;
}
inline void build(int k,int rt,int l,int r)
{
if (l==r)
{
tree[rt][k]=dep[num[l]][k];
return;
}
int mid=l+r>>1;
build(k,rt<<1,l,mid); build(k,rt<<1|1,mid+1,r);
tree[rt][k]=tree[rt<<1][k]+tree[rt<<1|1][k];
if (tree[rt][k]>=mod) tree[rt][k]-=mod;
}
inline int query(int k,int rt,int l,int r,int beg,int end)
{
if (l>=beg&&r<=end) return tree[rt][k];
int mid=l+r>>1,res=0;
if (beg<=mid) res+=query(k,rt<<1,l,mid,beg,end);
if (end>mid) res+=query(k,rt<<1|1,mid+1,r,beg,end);
if (res>=mod) res-=mod; return res;
}
inline void DFS1(int now,int fa,int d)
{
dep[now][1]=d; father[now]=fa; size[now]=1; int res=-1;
for (register int i=head[now];i!=-1;i=e[i].next)
if (e[i].to!=fa)
{
DFS1(e[i].to,now,d+1);
size[now]+=size[e[i].to];
if (size[e[i].to]>res) res=size[e[i].to],son[now]=e[i].to;
}
}
inline void DFS2(int now,int topf)
{
id[now]=++tot; num[tot]=now; top[now]=topf;
if (!son[now]) return; DFS2(son[now],topf);
for (register int i=head[now];i!=-1;i=e[i].next)
if (e[i].to!=father[now]&&e[i].to!=son[now]) DFS2(e[i].to,e[i].to);
}
inline void init(void)
{
for (register int i=1;i<=50;++i)
{
for (register int j=1;j<=n&&i!=1;++j)
dep[j][i]=(LL)dep[j][i-1]*dep[j][1]%mod;
build(i,1,1,n);
}
}
inline int get_sec(int k,int x,int y)
{
int ans=0;
while (top[x]!=top[y])
{
if (dep[top[x]][1]<dep[top[y]][1]) swap(x,y);
ans+=query(k,1,1,n,id[top[x]],id[x]);
if (ans>=mod) ans-=mod; x=father[top[x]];
}
if (dep[x][1]<dep[y][1]) swap(x,y);
ans+=query(k,1,1,n,id[y],id[x]);
if (ans>=mod) ans-=mod; return ans;
}
int main()
{
//freopen("CODE.in","r",stdin); freopen("CODE.out","w",stdout);
register int i; read(n);
memset(head,-1,sizeof(head));
memset(e,-1,sizeof(e));
for (i=1;i<n;++i)
read(x),read(y),add(x,y),add(y,x);
DFS1(rt,-1,0); DFS2(rt,rt); init(); read(m);
while (m--)
{
read(x); read(y); read(k);
write(get_sec(k,x,y)); putchar('\n');
}
return 0;
}
虽然100pts了,但是不解为什么过不了Hack数据。
算了不管明天用倍增这个传说中最稳定的算法搞一下,看看是不是我太非了。
Upt:倍增LCA不开O2轻松跑过......
我以后写LCA只用倍增!
CODE
#include<cstdio>
#include<cstring>
using namespace std;
typedef long long LL;
const int N=300005,K=55,P=25,mod=998244353;
struct edge
{
int to,next;
}e[N<<1];
int head[N],father[N][P],n,m,x,y,k,rt=1,cnt;
LL sum[N][K],dep[N][K];
inline char tc(void)
{
static char fl[100000],*A=fl,*B=fl;
return A==B&&(B=(A=fl)+fread(fl,1,100000,stdin),A==B)?EOF:*A++;
}
inline void read(int &x)
{
x=0; char ch=tc();
while (ch<'0'||ch>'9') ch=tc();
while (ch>='0'&&ch<='9') x=x*10+ch-'0',ch=tc();
}
inline void write(LL x)
{
if (x/10) write(x/10);
putchar(x%10+'0');
}
inline void add(int x,int y)
{
e[++cnt].to=y; e[cnt].next=head[x]; head[x]=cnt;
}
inline void swap(int &a,int &b)
{
int t=a; a=b; b=t;
}
inline void rebuild(int now)
{
for (register int i=2;i<=50;++i)
dep[now][i]=dep[now][i-1]*dep[now][1]%mod;
for (register int i=1;i<=50;++i)
sum[now][i]=(dep[now][i]+sum[father[now][0]][i])%mod;
}
inline void DFS(int now,int fa,int d)
{
father[now][0]=fa; dep[now][1]=d; rebuild(now);
for (register int i=head[now];i!=-1;i=e[i].next)
if (e[i].to!=fa) DFS(e[i].to,now,d+1);
}
inline void init(void)
{
for (register int j=0;j<P-1;++j)
for (register int i=1;i<=n;++i)
if (father[i][j]) father[i][j+1]=father[father[i][j]][j];
}
inline int LCA(int x,int y)
{
if (dep[x][1]<dep[y][1]) swap(x,y);
for (register int i=P-1;i>=0;--i)
if (father[x][i]&&dep[father[x][i]][1]>=dep[y][1]) x=father[x][i];
if (!(x^y)) return x;
for (register int i=P-1;i>=0;--i)
if (father[x][i]&&father[y][i]&&father[x][i]!=father[y][i])
{
x=father[x][i];
y=father[y][i];
}
return father[x][0];
}
int main()
{
//freopen("CODE.in","r",stdin); freopen("CODE.out","w",stdout);
register int i; read(n);
memset(head,-1,sizeof(head));
memset(e,-1,sizeof(e));
for (i=1;i<n;++i)
read(x),read(y),add(x,y),add(y,x);
DFS(rt,0,0); init(); read(m);
while (m--)
{
read(x); read(y); read(k);
int fa=LCA(x,y);
write((sum[x][k]+sum[y][k]+(LL)2*mod-sum[fa][k]-sum[father[fa][0]][k])%mod); putchar('\n');
}
return 0;
}
Luogu P4427 [BJOI2018]求和的更多相关文章
- P4427 [BJOI2018]求和
P4427 [BJOI2018]求和 同[TJOI2018]教科书般的扭曲虚空 懒得写了(雾 #include<bits/stdc++.h> #define il inline #defi ...
- 洛谷P4427 [BJOI2018]求和
\(\Large\textbf{Description: } \large{一颗n个节点的树,m次询问,每次查询点i到点j的路径上所有节点点深度的k次方的和并对998244353取模(1\leq n, ...
- 【BZOJ5293】[BJOI2018]求和(前缀和,LCA)
[BZOJ5293][BJOI2018]求和(前缀和,LCA) 题面 BZOJ 洛谷 题解 送分题??? 预处理一下\(k\)次方的前缀和. 然后求个\(LCA\)就做完了?... #include& ...
- bzoj5293: [Bjoi2018]求和
题目链接 bzoj5293: [Bjoi2018]求和 题解 暴力 对于lca为1的好坑啊.... 代码 #include<cmath> #include<cstdio> #i ...
- LCA+差分【p4427】[BJOI2018]求和
Description master 对树上的求和非常感兴趣.他生成了一棵有根树,并且希望多次询问这棵树上一段路径上所有节点深度的\(k\) 次方和,而且每次的\(k\) 可能是不同的.此处节点深度的 ...
- BZOJ5293: [Bjoi2018]求和 树上差分
Description master 对树上的求和非常感兴趣.他生成了一棵有根树,并且希望多次询问这棵树上一段路径上所有节点深度的k 次方和,而且每次的k 可能是不同的.此处节点深度的定义是这个节点 ...
- 【刷题】BZOJ 5293 [Bjoi2018]求和
Description master 对树上的求和非常感兴趣.他生成了一棵有根树,并且希望多次询问这棵树上一段路径上所有节点深度的k 次方和,而且每次的k 可能是不同的.此处节点深度的定义是这个节点到 ...
- BZOJ5293:[BJOI2018]求和(LCA,差分)
Description master 对树上的求和非常感兴趣.他生成了一棵有根树,并且希望多次询问这棵树上一段路径上所有节点深度的k 次方和,而且每次的k 可能是不同的.此处节点深度的定义是这个节点 ...
- 【Bzoj4555】【Luogu P4091】求和(NTT)
题面 Bzoj Luogu 题解 先来颓柿子 $$ \sum_{i=0}^n\sum_{j=0}^iS(i,j)2^jj! \\ =\sum_{j=0}^n2^jj!\sum_{i=0}^nS(i,j ...
随机推荐
- tab 页形式展现多张报表
业务系统中,很多报表都是沿用之前 EXCEL 的报表样式,原来以 sheet 格式显示的表,客户在 web 端展现的时候也希望也有同样的格式,润乾在实现这种效果和 EXCEL 一样简单灵活,轻松将数据 ...
- Apache POI导出excel表格
项目中我们经常用到导出功能,将数据导出以便于审查和统计等.本文主要使用Apache POI实现导出数据. POI中文文档 简介 ApachePOI是Apache软件基金会的开放源码函式库,POI提供A ...
- asp.net根据参数找不到记录后响应404及显示错误页
在asp.net mvc 中,action方法里根据参数获取数据,假如获取的数据为空,为了响应404错误页,我们可以return HttpNotFound(); 但是在asp.net webform中 ...
- CRM 各种类型字段的 赋值 取值
//lookupRecordEntity["new_lead"] = new EntityReference(entity.LogicalName, entity.Id); 获取时 ...
- Solr全文检索
1.Solr的安装 .环境要求 jdk1.+tomcat8+solr5.5.0 .将以上的软件包上传到服务器 . 安装tomcat8(解压文件) tar -zxvf apache-tomcat-.ta ...
- Windows Server 2012 RS 配置IIS8.0+发布网站
一.配置iis 8.0 IIS 8.0 是 windows server 2012 自带的服务器管理系统.相比之前版本,IIS 8.0 安装和操作都更加简单,界面也很简洁,安装也很迅速. 1. 进入w ...
- JS学习小结(上)
学而时习之,不亦说乎,开启JS学习新乐章~ JS是干啥的?网页特效,它主要是实现控制结构和样式,是一种行为,有多重要,不言而喻吧,页面炫酷的资本. 1. JS输出: alert("hello ...
- HBase——强一致性详解
Hbase是一个强一致性数据库,不是“最终一致性”数据库,官网给出的介绍: “Strongly consistent reads/writes: HBase is not an "event ...
- 一个比较全面 的web项目实战学习
一个比较全面 的web项目实战学习:http://www.cnblogs.com/jikey/p/3613082.html
- XML Namespace (xmlns) 属性
http://www.w3school.com.cn/xml/xml_namespaces.asp XML Namespace (xmlns) 属性 XML 命名空间属性被放置于元素的开始标签之中,并 ...