NOIP2023模拟16联测37 大眼鸹猫

题目大意

有两个长度为 n n n的序列 a , b a,b a,b,这两个序列都是单调不降的。

你可以对 a a a进行不超过 m m m次操作,每次操作你可以选择一个 i i i满足 1 ≤ i ≤ n 1\leq i\leq n 1≤i≤n,然后选择一个整数(可以是负数) x x x,将 a i a_i ai加上 x x x,这次操作要花费 x 2 x^2 x2的代价。

在操作的过程中,你需要保证 a a a始终单调不降。

最后,你需要将 a a a序列变为 b b b序列,即对任意 i i i满足 1 ≤ i ≤ n 1\leq i\leq n 1≤i≤n,都有 a i = b i a_i=b_i ai=bi。

求需要花费的总代价的最小值。输出答案模 998244353 998244353 998244353后的值。如果不可能让 a a a序列变为 b b b序列,则输出 − 1 -1 −1。

1 ≤ n , m ≤ 1 0 5 , 0 ≤ a i , b i ≤ 1 0 9 1\leq n,m\leq 10^5,0\leq a_i,b_i\leq 10^9 1≤n,m≤105,0≤ai,bi≤109

题解

首先,我们可以对每个值不等于 b i b_i bi的 a i a_i ai都加上 b i − a i b_i-a_i bi−ai。我们发现,对于相邻的两个位置,我们可以规定一个修改的先后顺序以满足在修改的时候这两个位置始终保持前一个位置的 a a a值不超过后一个位置的 a a a值。那么,因为这些先后顺序不会有环,所以这样是可以保证 a a a始终单调不降的。

我们按上面的方法操作,如果操作次数不够就输出 − 1 -1 −1。

如果还剩下一些操作次数,则我们可以用这些操作次数来减少代价。

根据基本不等式,我们将一个 + x +x +x尽量平均地分成多次加法能使代价最少。那么,对于每个位置要加的 x x x,我们记录它当前被拆成了多少份,设拆成了 k k k份,我们求出将其拆成 k + 1 k+1 k+1份相比于 k k k份能将代价减少多少。一开始每种操作都可以看作被拆成一份,我们以减少的代价为关键字来将这些操作放在大根堆里,每次取出堆顶并将代价减少对应的量,然后更新这次操作被拆成的数 k k k,即令 k = k + 1 k=k+1 k=k+1,然后继续算出将其拆成 k + 1 k+1 k+1份相比于 k k k份能将代价减少多少并放入堆中,这样将剩下的操作都用完,即可得出答案。

时间复杂度为 O ( ( n + m ) log ⁡ n ) O((n+m)\log n) O((n+m)logn)。

code

cpp 复制代码
#include<bits/stdc++.h>
using namespace std;
const int N=100000;
const long long mod=998244353;
int n,m;
long long ans=0,a[N+5],b[N+5];
struct node{
	long long x,k,v;
	bool operator<(const node ax)const{
		return v<ax.v;
	}
};
priority_queue<node>q;
long long dv(long long x,long long k){
	return (x/k)*(x/k)*(k-x%k)+(x/k+1)*(x/k+1)*(x%k);
}
long long gt(long long x,long long k){
	return dv(x,k)-dv(x,k+1);
}
int main()
{
//	freopen("attend.in","r",stdin);
//	freopen("attend.out","w",stdout);
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++) scanf("%lld",&a[i]);
	for(int i=1;i<=n;i++) scanf("%lld",&b[i]);
	for(int i=1;i<=n;i++){
		if(a[i]!=b[i]){
			--m;
			ans=(ans+(a[i]-b[i])*(a[i]-b[i])%mod)%mod;
			q.push((node){abs(a[i]-b[i]),1,gt(abs(a[i]-b[i]),1)});
		}
	}
	if(m<0){
		printf("-1");
		return 0;
	}
	if(q.empty()){
		printf("%lld",ans);
		return 0;
	}
	while(m--){
		node t=q.top();q.pop();
		ans=(ans-t.v%mod+mod)%mod;
		q.push((node){t.x,t.k+1,gt(t.x,t.k+1)});
	}
	printf("%lld",ans);
	return 0;
}
相关推荐
奋斗的小花生2 小时前
c++ 多态性
开发语言·c++
闲晨2 小时前
C++ 继承:代码传承的魔法棒,开启奇幻编程之旅
java·c语言·开发语言·c++·经验分享
UestcXiye3 小时前
《TCP/IP网络编程》学习笔记 | Chapter 3:地址族与数据序列
c++·计算机网络·ip·tcp
霁月风4 小时前
设计模式——适配器模式
c++·适配器模式
jrrz08285 小时前
LeetCode 热题100(七)【链表】(1)
数据结构·c++·算法·leetcode·链表
咖啡里的茶i5 小时前
Vehicle友元Date多态Sedan和Truck
c++
海绵波波1075 小时前
Webserver(4.9)本地套接字的通信
c++
@小博的博客5 小时前
C++初阶学习第十弹——深入讲解vector的迭代器失效
数据结构·c++·学习
爱吃喵的鲤鱼6 小时前
linux进程的状态之环境变量
linux·运维·服务器·开发语言·c++
7年老菜鸡7 小时前
策略模式(C++)三分钟读懂
c++·qt·策略模式