题解 [HNOI2019]序列
题目大意
给出一个\(n\)个数的数列\(A_{1,2,...,n}\),求出一个单调不减的数列\(B_{1,2,...,n}\),使得\(\sum_{i=1}^{n}(A_i-B_i)^2\)最小。
有\(m\)次查询,每次将某个\(A_x\)更改为\(y\),求出修改后的答案。查询之间互相独立。
\(n,m\le 10^5\)
思路
其实这道题正解是用保序回归,但是找规律也能找出来。
我们通过观察发现,对于一段相同的\(B_i\),\(B_i\)是该段的平均值。于是我们大胆猜测,我们最优方案就是把\(A\)序列划分成一些区间,每一段的\(B\)都是\(A\)的平均值,并且\(B\)单调不降。
我们发现这顺便还可以发现一个事情:我们肯定应该多分区间,否则的话我们肯定只分一段就完事了。
于是,我们现在考虑一个区间对答案的贡献:
\]
其中\(d\)是这段区间的平均值。
\]
\]
于是我们发现我们只需要维护区间平方和、区间和、区间长度。于是,我们就顺利地拿到了\(50\)分。
考虑\(100\)分。一个很显然的事情是,我们会改变的决策区间一定是\([L,R]\),至于\([1,L)\)和\((R,n]\)都不会被影响,于是我们可以预处理一下。
问题就是如何找到\(L,R\)。一个不是很显然的事情就是,我们选的端点一定都是一开始分的区间的某些端点。因为我们一个区间如果从中间分开,前一段的平均值一定比后一段的平均值大,因为如果比它小的话肯定分开更优(分得越多越优)。于是,如果我们不是选一开始的端点的话,被分开的那一段前一段一定比后一段的平均值大,与平均值单调不减矛盾,得证。
同时,我们还发现答案是具有单调性的。于是我们可以考虑先二分\(R\),然后再二分\(L\)。二分\(R\)直接二分就好了,二分\(L\)的话,我们发现并不需要考虑前面的是否满足条件(这个自己想一下就明白了),只需要考虑分开的后面的,于是这个我们可以在主席树上查询。具体见代码。
时间复杂度\(\Theta(n\log^2 n)\),空间复杂度\(\Theta(n\log n)\)。
\(\texttt{Code}\)
#include <bits/stdc++.h>
using namespace std;
#define Int register int
#define mod 998244353
#define ll long long
#define MAXN 100005
template <typename T> inline void read (T &t){t = 0;char c = getchar();int f = 1;while (c < '0' || c > '9'){if (c == '-') f = -f;c = getchar();}while (c >= '0' && c <= '9'){t = (t << 3) + (t << 1) + c - '0';c = getchar();} t *= f;}
template <typename T,typename ... Args> inline void read (T &t,Args&... args){read (t);read (args...);}
template <typename T> inline void write (T x){if (x < 0){x = -x;putchar ('-');}if (x > 9) write (x / 10);putchar (x % 10 + '0');}
int n,m,a[MAXN],inv[MAXN];
int mul (int a,int b){return 1ll * a * b % mod;}
int dec (int a,int b){return a >= b ? a - b : a + mod - b;}
int add (int a,int b){return a + b >= mod ? a + b - mod : a + b;}
struct node{
int rt,ans,top;//对应主席树的顶点、答案、栈顶编号
}pre[MAXN],suf[MAXN];
struct Data{
int len,sqr;ll sum;//因为要比较平均值大小,所以区间和不能取模
Data(){}
Data(int len1,ll sum1,int sqr1){len = len1,sum = sum1,sqr = sqr1;}
Data operator + (Data p){return Data(len + p.len,sum + p.sum,add (sqr,p.sqr));}
Data operator - (Data p){return Data(len - p.len,sum - p.sum,dec (sqr,p.sqr));}
bool operator < (Data p)const{return p.len ? (len ? 1.0 * sum / len < 1.0 * p.sum / p.len: 1) : 0;}
bool operator <= (Data &p)const{return p.len ? (len ? 1.0 * sum / len <= 1.0 * p.sum / p.len: 1) : 0;}
int calc (){return dec (sqr,mul (mul (sum % mod,sum % mod),inv[len]));}
}sum[MAXN],sta[MAXN];
Data calc (int l,int r){return sum[r] - sum[l - 1];}
struct Segment{
int cnt;
struct Node{
int son[2],l,r,k;//k是指l->r这写区间第一段区间的右端点
}tree[MAXN * 60];
void Pushup (int x){
tree[x].l = tree[tree[x].son[0]].l,tree[x].r = tree[tree[x].son[tree[x].son[1] > 0]].r;
tree[x].k = tree[tree[x].son[0]].k;
}
void modify (int &x,int l,int r,int pos,int L,int R){
tree[++ cnt] = tree[x],x = cnt;
if (l == r) return tree[x].l = L,tree[x].r = tree[x].k = R,void ();
int mid = (l + r) >> 1;
if (pos <= mid) modify (tree[x].son[0],l,mid,pos,L,R);
else modify (tree[x].son[1],mid + 1,r,pos,L,R);
Pushup (x);
}
int queryr (int x,int l,int r,int pos){//查找第pos段区间的右端点
if (l == r) return tree[x].r;
int mid = (l + r) >> 1;
if (pos <= mid) return queryr (tree[x].son[0],l,mid,pos);
else return queryr (tree[x].son[1],mid + 1,r,pos);
}
int queryl (int x,int l,int r,int pos,Data &tmp){//找到最靠右的满足的L,并对答案进行合并
if (r <= pos){
Data Lget = calc (tree[x].l,tree[x].k),Rget = calc (tree[x].k + 1,tree[x].r);
if (Rget + tmp <= Lget) return tmp = tmp + calc (tree[x].l,tree[x].r),0;
if (l == r) return tree[x].r;
}
int res,mid = (l + r) >> 1;
if (pos > mid && (res = queryl (tree[x].son[1],mid + 1,r,pos,tmp))) return res;
else return queryl (tree[x].son[0],l,mid,pos,tmp);
}
}Tree;
void Init(){
for (Int i = 1,top = 0;i <= n;++ i){
sta[++ top] = Data (1,a[i],mul (a[i],a[i]));
while (top > 1 && sta[top] <= sta[top - 1]) sta[top - 1] = sta[top - 1] + sta[top],-- top;
pre[i].ans = add (pre[i - sta[top].len].ans,sta[top].calc());//计算答案
pre[i].rt = pre[i - 1].rt,pre[i].top = top;
Tree.modify (pre[i].rt,1,n,top,i - sta[top].len + 1,i);
}
for (Int i = n,top = 0;i;-- i){
sta[++ top] = Data (1,a[i],mul (a[i],a[i]));
while (top > 1 && sta[top - 1] <= sta[top]) sta[top - 1] = sta[top - 1] + sta[top],-- top;
suf[i].ans = add (suf[i + sta[top].len].ans,sta[top].calc());
suf[i].rt = suf[i + 1].rt,suf[i].top = top;
Tree.modify (suf[i].rt,1,n,top,i,i + sta[top].len - 1);
}
}
signed main(){
read (n,m),inv[1] = 1;
for (Int i = 2;i <= n;++ i) inv[i] = mul (mod - (mod / i),inv[mod % i]);
for (Int i = 1;i <= n;++ i) read (a[i]),sum[i] = sum[i - 1] + Data (1,a[i],mul (a[i],a[i]));Init ();
write (pre[n].ans),putchar ('\n');
for (Int i = 1,x,y;i <= m;++ i){
read (x,y);
int l = 0,r = suf[x + 1].top - 1;
while (l <= r){
int mid = (l + r) >> 1,Rpos = mid ? Tree.queryr (suf[x + 1].rt,1,n,suf[x + 1].top - mid + 1) : x;//Rpos就是选出来的R
Data tmp = Data (1,y,mul (y,y)) + calc (x + 1,Rpos);int Lpos = x > 1 ? Tree.queryl (pre[x - 1].rt,1,n,pre[x - 1].top,tmp) : x;
if (tmp < calc (Rpos + 1,Tree.queryr (suf[x + 1].rt,1,n,suf[x + 1].top - mid))) r = mid - 1;
else l = mid + 1;
}
int mid = r + 1,Rpos = mid ? Tree.queryr (suf[x + 1].rt,1,n,suf[x + 1].top - mid + 1) : x;
Data tmp = Data (1,y,mul (y,y)) + calc (x + 1,Rpos);int Lpos = x > 1 ? Tree.queryl (pre[x - 1].rt,1,n,pre[x - 1].top,tmp) : x;
write (add (tmp.calc(),add (pre[Lpos].ans,suf[Rpos + 1].ans))),putchar ('\n');
}
return 0;
}
题解 [HNOI2019]序列的更多相关文章
- 题解-[HNOI2016]序列
题解-[HNOI2016]序列 [HNOI2016]序列 给定 \(n\) 和 \(m\) 以及序列 \(a\{n\}\).有 \(m\) 次询问,每次给定区间 \([l,r]\in[1,n]\),求 ...
- 【题解】Luogu P5294 [HNOI2019]序列
原题传送门 题意:给你一个长度为\(n\)的序列\(A\),每次询问修改一个元素(只对当前询问有效),然后让你找到一个不下降序列\(B\),使得这两个序列相应位置之差的平方和最小,并输出这个最小平方和 ...
- [题解] [SDOI2017] 序列计数
题面 题解 和 SDOI2015 序列统计 比较像 这个无非就是把乘改成了加, NTT 改成了 MTT 再加上了一个小小的容斥 : 拿所有方案减去不合法方案即可 Code #include <a ...
- [题解] [SDOI2015] 序列统计
题面 题解 设 \(f[i][j]\) 代表长度为 \(i\) 的序列, 乘积模 \(m\) 为 \(j\) 的序列有多少个 转移方程如下 \[ f[i + j][C] = \sum_{A*B\equ ...
- luogu P5294 [HNOI2019]序列
传送门 这个什么鬼证明直接看uoj的题解吧根本不会证明 首先方案一定是若干段等值的\(B\),然后对于一段,\(B\)的值应该是\(A\)的平均值.这个最优方案是可以线性构造的,也就是维护以区间平均值 ...
- 题解-bzoj1283序列 & bzoj4842 [Neerc2016]Delight for a Cat
因为这两题有递进关系,所以放一起写 Problem bzoj1283 序列 题意概要:一个长度为 \(n\) 的序列\(\{c_i\}\),求一个子集,使得原序列中任意长度为 \(m\) 的子串中被选 ...
- 【洛谷5294】[HNOI2019] 序列(主席树维护单调栈+二分)
点此看题面 大致题意: 给你一个长度为\(n\)的序列\(A\),每次询问修改一个元素(只对当前询问有效),然后让你找到一个不下降序列\(B\),使得这两个序列相应位置之差的平方和最小,并输出这个最小 ...
- 【Luogu5294】[HNOI2019]序列
题目链接 题意 给定一个序列,要求将它改造成一个非降序列,修改一个数的代价为其改变量的平方. 最小化总代价. 另有\(Q\) 次询问,每次修改一个位置上的数.(询问之间独立,互不影响) Sol 神仙 ...
- bzoj1211树的计数 x bzoj1005明明的烦恼 题解(Prufer序列)
1211: [HNOI2004]树的计数 Time Limit: 10 Sec Memory Limit: 162 MBSubmit: 3432 Solved: 1295[Submit][Stat ...
随机推荐
- 新东方APP技术架构演进, 分布式系统架构经验分享
今天的演讲题目是"新东方APP技术架构演进, C端技术经验分享" 作者:张建鑫, 曾任IBM高级软件架构师, 滴滴高级技术专家, 现任新东方集团高级技术总监 古代东西方的思想家都产 ...
- Dockerfile 实践及梳理
Dockerfile 是一个文本文件,我们可以通过组合一条条的指令 (Instruction),来构建满足我们需求的 Docker 镜像 文档 Best practices for writing D ...
- 初步测试VS2017+Win10IoT,基于World Map of Makers
测试World Map of Makers Demo,创建一个简单的internet连接应用程序来查询web API的状态(目测是一个延时时间),并使用该信息切换LED. 官方仓库的Solution是 ...
- 每天迁移MySQL历史数据到历史库Python脚本
#!/usr/bin/env python # coding:utf-8 #__author__ = 'Logan' import MySQLdb import sys import dat ...
- Windows内核基础知识-8-监听进程、线程和模块
Windows内核基础知识-8-监听进程.线程和模块 Windows内核有一种强大的机制,可以在重大事件发送时得到通知,比如这里的进程.线程和模块加载通知. 本次采用链表+自动快速互斥体来实现内核的主 ...
- for循环练习之打印三角形
public class TestDemo01 { /** * 打印三角形 * 1.打印空格 * 2.打印三角形 */ public static void main(String[] args) { ...
- 新版idea无法导入mavenweb模板
目前没有任何办法,最好是下个旧版的
- 为开源项目 go-gin-api 增加后台任务模块
目录 任务管理界面 (WEB) 任务调度器 任务执行器 小结 推荐阅读 任务管理界面 (WEB) 支持在 WEB 界面 中对任务进行管理,例如:新增任务.编辑任务.启用/禁用任务.手动执行任务 等. ...
- mybaits源码分析--事务管理(八)
一.事务管理 写到这也快进入收尾阶段了了,在介绍MyBatis中的事务管理时不可避免的要接触到DataSource的内容,所以接下来会分别来介绍DataSource和Transaction两块内容. ...
- Systemd Journald占用资源过多
journald占用过多磁盘空间 方法一 检查当前journal使用磁盘量 journalctl --disk-usage 清理方法可以采用按照日期清理,或者按照允许保留的容量清理,只保存2天的日志, ...