题解:P2263 命运的彼方

题意

给定序列 hhh,求 min⁡x(min⁡l,l≤n−k+1∑i=ll+k−1∣x−hi∣)\min\limits_x\left(\min\limits_{l,l\le n-k+1} \sum_{i = l}^{l+k-1} |x - h_i|\right)xmin(l,l≤n−k+1min∑i=ll+k−1∣x−hi∣)。

做法

如果我有一个长度为 kkk 的序列 aaa,任取一个 xxx 求 ∑i=1k∣x−ai∣\sum_{i = 1}^k |x - a_i|∑i=1k∣x−ai∣,xxx 肯定取 aaa 的中位数。这非常简单,我们可以想象在坐标轴上有 kkk 个点,我们设定 xxx 在最左边,那么可以算出距离总和 ansansans。xxx 加 111,相当于 ans+=左边点数−右边点数ans += \text{左边点数} - \text{右边点数}ans+=左边点数−右边点数。那么 xxx 到达正中央(或中央两点之间)前,ansansans 单调递减(左边点数 <<< 右边点数),之后又单调递增(左边点数 >>> 右边点数)。所以当 xxx 为中位数时,ansansans 最小。

考虑枚举区间右端点 rrr,现在的问题是如何快速求出区间 (l,r)(l,r)(l,r) 的中位数以及答案。有很多种数据结构都可以在 O(log⁡n)O(\log n)O(logn) 复杂度内加入/删除一个数(因为显然要离散化),树状数组最好写,就用它好了。设一个区间 (l,r)(l,r)(l,r) 的中位数是 midmidmid,区间中小于 midmidmid 的数的个数是 cntcntcnt,小于 midmidmid 的数的和是 sumsumsum,区间的和是 uuu,区间的答案就是 2×mid×cnt−2×sum+u−k×mid2 \times mid \times cnt - 2 \times sum + u - k \times mid2×mid×cnt−2×sum+u−k×mid,随意计算并合并同类项即可。所以需要维护两个树状数组,一个维护一段前缀内数的个数,另一个维护一段前缀内数字的和。最大的问题是如何找到中位数。可以利用倍增的思想找中位数(利用树状数组的结构),做到 O(log⁡n)O(\log n)O(logn)。

代码

时间复杂度 O(nlog⁡n)O(n \log n)O(nlogn),1.11s。

cpp 复制代码
#include <bits/stdc++.h>
using namespace std;
#define rep(i,n) for(int i = 1;i <= n;++i)
#define pre(i,n) for(int i = n;~i;--i)
#define int long long
#define swap(x,y) (x ^= y ^= x ^= y)
#define debug cerr<<"Help!\n"
constexpr int N = 5e5 + 5,inf = 1e18;
int h[N],b[N],val[N],tot,n,k,pos,res = inf,u,mid,cnt,sum,tmp;
struct Fen{//树状数组
	#define lowbit(x) (x & -x)
	int c[N];
	inline void add(int x,int ad){
		for(;x <= N - 5;x += lowbit(x)) c[x] += ad;
	}
	inline int query(int x){
		int res = 0;
		for(;x;x ^= lowbit(x)) res += c[x];
		return res;
	}
	inline int find(int k){//找到树状数组内第 k 大
		int res = 0,nxt;
		pre(i,20){
			nxt = res + (1 << i);
			if(nxt <= N - 5 && c[nxt] < k) res = nxt,k -= c[nxt];
		}
		return res + 1;
	}
}fen[2];//fen[0] 维护数的个数,fen[1] 维护和
signed main(){
	ios::sync_with_stdio(false);
	cin.tie(0),cout.tie(0);
	cin>>n>>k,pos = (k + 1) >> 1;
	rep(i,n) cin>>h[i],b[i] = h[i];
	sort(b + 1,b + n + 1);
	tot = unique(b + 1,b + n + 1) - b - 1;//离散化
	rep(i,n){
		tmp = h[i],h[i] = lower_bound(b + 1,b + tot + 1,h[i]) - b;
		val[h[i]] = tmp;
		fen[0].add(h[i],1),fen[1].add(h[i],val[h[i]]),u += val[h[i]];//加入这个数
		if(i < k) continue;
		if(i ^ k) fen[0].add(h[i-k],-1),fen[1].add(h[i-k],-val[h[i-k]]),u -= val[h[i-k]];//删除不在区间内的数
		mid = fen[0].find(pos),cnt = fen[0].query(mid - 1),sum = fen[1].query(mid - 1);
		tmp = 2 * val[mid] * cnt - 2 * sum + u - k * val[mid];
		res = min(res,tmp);
	}
	cout<<res;
	cerr<<'\n'<<1.0 * clock() / CLOCKS_PER_SEC;
	return 0;
}
相关推荐
Prince-Peng6 小时前
技术架构系列 - 详解Redis
数据结构·数据库·redis·分布式·缓存·中间件·架构
码农水水7 小时前
得物Java面试被问:消息队列的死信队列和重试机制
java·开发语言·jvm·数据结构·机器学习·面试·职场和发展
-Try hard-7 小时前
数据结构:链表常见的操作方法!!
数据结构·算法·链表·vim
wengqidaifeng7 小时前
数据结构---顺序表的奥秘(下)
c语言·数据结构·数据库
嵌入小生0077 小时前
单向链表的常用操作方法---嵌入式入门---Linux
linux·开发语言·数据结构·算法·链表·嵌入式
千谦阙听7 小时前
数据结构入门:栈与队列
数据结构·学习·visual studio
定偶7 小时前
MySQL知识点
android·数据结构·数据库·mysql
dazzle8 小时前
Python数据结构(十二):插入排序详解
数据结构·python·排序算法
希望有朝一日能如愿以偿8 小时前
力扣每日一题
数据结构·算法·leetcode
草履虫建模8 小时前
力扣算法分析 27.移除元素
java·开发语言·数据结构·后端·算法·leetcode·排序算法