[ABC376E] Max × Sum 题解

合集 - 题解(15)

1.[ABC347C] Ideal Holidays题解2024-03-312.《扶苏的问题》题解2024-06-023.[ABC366C] Balls and Bag Query 题解2024-08-124.[ABC366D] Cuboid Sum Query 题解2024-08-125.P7706 文文的摄影布置 题解2024-08-216.[ABC370C] Word Ladder 题解2024-09-087.P11020 「LAOI-6」Radiation 题解2024-09-088.P11019 「LAOI-6」[太阳]] 请使用最新版手机 QQ 体验新功能 题解2024-09-089.[ABC371D] 1D Country 题解2024-09-1510.[ABC371D] 1D Country 线段树解法2024-09-1511.P1955 程序自动分析 题解2024-10-18

12.[ABC376E] Max × Sum 题解2024-10-20

13.[ABC379D] Home Garden 题解2024-11-1014.P11276 第一首歌 题解2024-11-1515.[ABC380C] Move Segment 题解2024-11-18

收起

[ABC376E] Max × Sum 题解

原题链接

洛谷链接

一道简单的推性质题，首先明确一个性质，子集是非连续的，所以在计算时并不用连续区间求。

拿过题来，首先想的是枚举 $B$ 的最小子集，但其复杂度为 $O(C_N^K)$ 复杂度过高，不足以通过本题。于是转变思路，枚举 $A$ 之中的最大值。若 $a_i$ 是一个区间最大值，当且仅当长度为 $k$ 的子集其为最大。但这样还不是很好求。

由于题目未要求输出子集元素，则元素顺序对求解过程无影响，于是建立结构体数组 $s$ 存储 $A,B$ ，则有 $s_i=\{a_i,b_i\}$ ，以 $a_i$ 的大小进行结构体排序，从大到小遍历数组，建立大根堆维护 $b_i$ 的，若堆的大小超出 $k$ 则弹出堆顶，和减去。对于每一个堆的大小等于 $k$ 的时刻，统计答案。（详见代码）

注：

1. 记得开 long long
2. ans 的初值不能太小，至少 $2e17$

#include <bits/stdc++.h>
#define seq(q, w, e) for (int q = w; q <= e; q++)
#define ll long long              //记得开long long
using namespace std;
const ll maxn = 2e5+10,mod=2e17;  //初值至少2e17
ll t,ans;
ll n,k,tot;                       //tot为b的和
struct node{
	ll a,b;
}s[maxn];
bool cmp(node a,node b){         //排序函数
	return a.a<b.a;
}
signed main(){
    ios::sync_with_stdio(0);
    cin.tie(0);
    cin>>t;
	while(t--){
		tot=0,ans=mod;
		priority_queue<ll,vector<ll>,less<ll> >pq; //优先队列存储b的值
		cin>>n>>k;
		seq(i,1,n) cin>>s[i].a;
		seq(i,1,n) cin>>s[i].b;
		sort(s+1,s+n+1,cmp);
		seq(i,1,n){
			if(pq.size()<k-1){                    //若堆的大小不足k-1
				tot+=s[i].b;
				pq.push(s[i].b);
			}
			else{
				tot+=s[i].b;                      //压入，正好k个元素
				pq.push(s[i].b);
				ans=min(ans,s[i].a*tot);          //与原先ans取min
				tot-=pq.top();                    //不确定是否最优，于是继续统计
				pq.pop();
			}
		}
		cout<<ans<<"\n";
	}
    return 0;
}