正题

题目链接:https://www.luogu.com.cn/problem/P6834?contestId=34123

题目大意

nnn个点，每个点i(i≠1)i(i\neq 1)i(i=1)随机连接[i−k,i−1][i-k,i-1][i−k,i−1]中的一个点。对于每个点有一个权值aia_iai。每次可以选择一个全部都是正权值的连通块里所有权值减一，求最优期望次数。

解题思路

考虑每个点的贡献，对于一个节点xxx和它的父节点fafafa显然根据铺设道路那题的做法，这个点的贡献就是max{ax−afa,0}max\{a_x-a_{fa},0\}max{ax−afa,0}

所以我们需要维护一个数据结构可以查询比axa_xax小的数的和，支持插入删除。显然离散化之后用树状数组即可。

时间复杂度O(nlog⁡n)O(n\log n)O(nlogn)

codecodecode

#include<cstdio>
#include<cstring>
#include<algorithm>
#define lowbit(x) (x&-x)
#define ll long long
using namespace std;
const ll N=1e6+10,XJQ=998244353;
ll n,k,num,answer,cnt;
ll inv[N],a[N],b[N],t[N],z[N];
ll power(ll x,ll b){ll ans=1;while(b){if(b&1)ans=ans*x%XJQ;x=x*x%XJQ;b>>=1;}return ans;
}
void Change(ll x,ll val,ll flag){while(x<=cnt){t[x]=(t[x]+val)%XJQ;z[x]=z[x]+flag;x+=lowbit(x);}return;
}
ll Ask(ll x){ll ans=0;num=0;while(x){num=num+z[x];ans=(ans+t[x])%XJQ;x-=lowbit(x);}return ans;
}
int main()
{scanf("%lld%lld",&n,&k);inv[1]=1;for(ll i=2;i<=k;i++)inv[i]=XJQ-XJQ/i*inv[XJQ%i]%XJQ;for(ll i=1;i<=n;i++)scanf("%lld",&a[i]),b[++cnt]=a[i];sort(b+1,b+1+cnt);cnt=unique(b+1,b+1+cnt)-b-1;ll l=1,sum=0;answer=a[1];for(ll i=1;i<=n;i++){while(l<=n&&l<i-k){ll y=lower_bound(b+1,b+1+cnt,a[l])-b;Change(y,-a[l],-1);l++;}ll x=lower_bound(b+1,b+1+cnt,a[i])-b;ll w=Ask(x),ans=0;ans=(num*a[i]%XJQ-w+XJQ)%XJQ;answer=(answer+ans*inv[min(i-1,k)]%XJQ)%XJQ;Change(x,a[i],1);}printf("%lld",answer);
}

P6834-[Cnoi2020]梦原【数学期望,树状数组】相关推荐

BZOJ 2141 排队（块套树，分块，树状数组）【BZOJ修复工程】
整理的算法模板合集: ACM模板点我看算法全家桶系列!!! 实际上是一个全新的精炼模板整合计划题目链接 https://hydro.ac/d/bzoj/p/2141 是 hydro 的 BZOJ ...
树状数组思路以及性能分析
特点代码短.常数很小应用及时间复杂度区间查询:求前缀和单点修改:给某个位置上的数加上一个数(同时能以非常小的代价维护前缀和) 时间复杂度:O(logn)O(logn)O(logn) 与一般前缀 ...
HDU 5869 Different GCD Subarray Query 树状数组 + 一些数学背景
http://acm.hdu.edu.cn/showproblem.php?pid=5869 题意:给定一个数组,然后给出若干个询问,询问[L, R]中,有多少个子数组的gcd是不同的. 就是[L, ...
Codeforces数学1600----day1[同余定理，树状数组+两次二分，，组合计数]
1.C. Kuroni and Impossible Calculation **知识点:同余定理 ** #include <iostream> #include <cstdio&g ...
【HDU - 5869】Different GCD Subarray Query（思维，数学，gcd，离线处理，查询区间不同数，树状数组或二分RMQ）
题干: This is a simple problem. The teacher gives Bob a list of problems about GCD (Greatest Common Di ...
[CF808B] Average Sleep Time（[强行]树状数组，数学）
题目链接:http://codeforces.com/contest/808/problem/B 题意:n个数,求其中每k个数的和的平均值. 鬼知道我怎么会石乐志上来就用了树状数组. 1 #inclu ...
数据结构--树状数组
文章目录 1. 树状数组 2. 单点修改 3. 区间修改 4. 完整代码 5. 参考文献 1. 树状数组类似数据结构:线段树(Segment Tree) 树状数组跟线段树的区别: 树状数组能做 ...
Codeforces 1096F(dp + 树状数组)
题目链接题意: 对于长度为$n$的排列,在已知一些位的前提下求逆序对的期望思路: 将答案分为$3$部分 $1.$$-1$与$-1$之间对答案的贡献.由于逆序对考虑的是数字之间的大小关系,故假设$- ...
吊打线段树的超级树状数组
你是否讨厌线段树那冗长的代码?你是否还在因为线段树的难调试而满头♂dark汗?那么,请不要错过!超级树状数组特价!只要998,只要998! ##￥--#--￥%--&%￥--ER#%$#$#^ ...

P6834-[Cnoi2020]梦原【数学期望,树状数组】

正题

题目大意

解题思路

codecodecode

P6834-[Cnoi2020]梦原【数学期望,树状数组】相关推荐

最新文章

热门文章