D Difference (二分 + 单调队列)

D Difference (二分 + 单调队列)

https://ac.nowcoder.com/acm/contest/34866/D

题意

给你长度为n的序列 和一个k
一个区段值满足: f(l, r) = (max - min) * (r - l + 1); 其中max min是区间最大最小值
要求输出第k大的区间值

思路

因为数据比较大 容易超时 可以考虑二分答案
然后去判断对于一个答案有几个大于等于它的 找到存在k个大于等于它的答案
（二分答案的复杂度是OlogV）
寻找一个大于等于ans的区间值的个数 可以考虑用单调队列
对于一个区间如果r固定[l, r]的区间值肯定比[l-1, r]大 因为[l, r]覆盖[l-1, r]前一个区间的区间长度比后一个大 而max-min肯定大于等于后一个区间
而要求某个区间的最大最小值 可以用单调队列分别维护
固定有边界 然后每次找到左边界最右的且满足区间值等于当前二分的答案 的l 对于这个右边界 贡献就是l
遍历右边界1-n 将贡献累加起来 与k做比较即可

#include<bits/stdc++.h>
#include<unordered_map>
#include<algorithm>
#include<map>
#define ll long long
#define ull unsigned long long
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
using namespace std;
const ll inf = 0x3f3f3f3f;
const ll INF = 0x3f3f3f3f3f3f3f3f;
const double eps = 1e-4;
const ll N = 5e5 + 5;
const int M = 1e6 + 5;
const ll mod = 1e9 + 7;
ll n, a[N], k, que[N];

ll lmi, lmx, rmi, rmx;
ll quemi[N], quemx[N];//储存队列中第i个元素在a数组中的下标

//判断函数
bool check(ll x) {
        //单调队列
	ll l = 1, ans = 0;
	lmi = 1, lmx = 1;//一开始左边初始化为1 右边初始化为0 便于第一次操作
	rmi = 0, rmx = 0;
	for (int i = 1; i <= n; i++) {
                //区间最大 最小值 队列头是最大或最小
                //每次判断队列尾是否比要放进来的值小或大 根据情况弹出队列 保证队列队头是最值后面是单调的
		while (lmi <= rmi && a[quemi[rmi]] >= a[i]) rmi--;
                //自己进队列
		quemi[++rmi] = i;
		while (lmx <= rmx && a[quemx[rmx]] <= a[i]) rmx--;
		quemx[++rmx] = i;

		while (l <= i) {
                        //判断当前区间是否是要找的那个区间
                        //如果区间值大于等于x 说明区间还可以再缩短
			if ((a[quemx[lmx]] - a[quemi[lmi]]) * (i - l + 1) >= x) {
                                //判断队头是否是l边界 如果是向右缩短的同时 队列队头弹出一个元素
				if (quemx[lmx] == l) lmx++;
				if (quemi[lmi] == l) lmi++;
				l++;
			}
			else break;
		}

		ans += l - 1;
	}
	if (ans >= k) return true;
	return false;
}

void solve() {
	cin >> n >> k;
	for (int i = 1; i <= n; i++) {
		cin >> a[i];
	}
        //二分答案
	ll l = 0, r = 1e15;
	ll ans = 0;
	while (l <= r) {
		ll mid = (l + r) / 2;
		if (check(mid)) {
			ans = mid;
			l = mid + 1;
		}
		else r = mid - 1;
	}

	cout << ans << "\n";
}
signed main() {
	IOS;
	int t = 1;
	//cin >> t;
	while (t--)
	{
		solve();
	}
}