【CF】1779C-Least Prefix Sum 题解

传送门：1779C
标签：数据结构

题目大意

给出一个长度为n的序列a₁,a₂,···,a_n和一个数m(1<=m<=n<=2e5)。定义i的前缀和为a₁+a₂+···+a_i，现在你需要使m的前缀和为序列所有前缀和中最小的，你可以进行如下操作：选择一个下标i，令a_i=-a_i，求满足要求所需要的最小操作数。
输入：T（测试组数），n，m，以及一个长度为n的序列
输出：最小操作数

算法分析

• 若从贪心的角度考虑，我们希望a_m及其之前的所有数都为负数，而其之后的数都为正数，这样前缀和的函数图像(y=a₁+a₂+···+a_x)就会呈"V"字形，而极小值点x=m自然也就符合题意。但这样做并不能保证操作数最小。其实我们只要让x=m为整个函数的最小值点，并不需要确保图像为"V"字形。
• 从数学的角度来看，函数的最小值点x₀满足：对于任意定义域中的x，都有f(x)-f(x₀)>=0。简单来说，只要序列中所有前缀和与m的前缀和之差都为非负数即可。此时可能会有人想：我每次都遍历所有前缀和，如果有比m前缀和小的，就对其中的最小数取相反数，反复操作直到m前缀和最小。然而这样做在最坏情况下时间复杂度是O(n²)，必T。我们应该往简单的方向思考，“前缀和”这个词往往和“差分”一起出现，而刚刚提到的f(x)-f(x₀)不就是差分d吗？当x<x₀时，d=a_x+1+a_x+2+···+a_x0，当x>x₀时，d=a_x0+1+a_x0+2+···+a_x，所以只要确保d>=0恒成立就符合题意。
• 现在我们已经知道要以m为起点分别往两边算差分，那么如果d<0我们要怎么处理呢？显然要从已经遍历过的数中找到一个最值，然后取其相反数。观察函数图像我们不难发现，若这种情况出现在m前面，我们要找最大值，否则找最小值。取最值的数据结构想必大家都不陌生——堆。STL库直接为我们提供了优先队列，十分的方便。也就是说，我们要在m前面建一个大根堆，在m后面建一个小根堆，值得注意的是，往m前面遍历的下标范围是[2,m]，m后面则是[m+1,n]。这样一来，即使在最坏的情况下时间复杂度也不会超过O(nlogn)，考虑到题目给的数据范围上限是2e5，显然是能过的。

代码实现

#include <iostream>
using namespace std;
#include <queue>
#include <vector>
priority_queue<int,vector<int>,less<int>> heap0;	//大根堆 
priority_queue<int,vector<int>,greater<int>> heap;	//小根堆 
int a[200001];
int main(){
	int i,j,x,cnt,n,m,T;
	long long sum;
	std::ios::sync_with_stdio(false);
	std::cin.tie(0);std::cout.tie(0);
	cin>>T;
	while(T--){
		cin>>n>>m;
		cnt=0;
		while(!heap0.empty())heap0.pop();//每组测试前清空 
		while(!heap.empty())heap.pop();
		for(i=1;i<=m-1;i++){
			cin>>a[i];
		}
		cin>>sum;
		if(m>1&&sum>0LL){	//第m个数必须为负数 
			cnt++;
			sum=-sum;
		}
		for(i=m-1;i>=2;i--){//往前遍历 
			if(sum+a[i]>0LL){
				cnt++;
				if((!heap0.empty())&&heap0.top()>a[i]){
					sum-=2*heap0.top();//更新差分 
					heap0.pop();	//取最大值 
					sum+=a[i];
					heap0.push(a[i]);
				}
				else sum-=a[i];
			}
			else if(a[i]>0){
				sum+=a[i];
				heap0.push(a[i]);
			}
			else sum+=a[i];	//负数无需入堆 
		}
		sum=0LL;
		for(i=m+1;i<=n;i++){//往后遍历 
			cin>>x;
			if(sum+x<0LL){
				cnt++;
				if((!heap.empty())&&heap.top()<x){
					sum-=2*heap.top();//更新差分 
					heap.pop();		//取最小值 
					sum+=x;
					heap.push(x);
				}
				else sum-=x;
			}
			else if(x<0){
				sum+=x;
				heap.push(x);
			}
			else sum+=x;	//正数无需入堆 
		}
		cout<<cnt<<'\n';	//最小操作数 
	}
	return 0;
}