题解: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;
}
相关推荐
努力努力再努力wz19 分钟前
【Linux网络系列】:TCP 的秩序与策略:揭秘传输层如何从不可靠的网络中构建绝对可靠的通信信道
java·linux·开发语言·数据结构·c++·python·算法
重生之后端学习4 小时前
78. 子集
java·数据结构·算法·职场和发展·深度优先
kronos.荒4 小时前
滑动窗口+哈希表:最小覆盖子串
数据结构·python·散列表
wostcdk4 小时前
筛质数汇总
数据结构·算法
qq_454245035 小时前
BuildTemplateGraph 函数深度解析:动态节点图构建的架构设计与核心价值
数据结构·c#
qq_454245036 小时前
SkeletonFlow:基于组合子逻辑与范畴论的数据流处理框架
数据结构·c#
王老师青少年编程6 小时前
2020年信奥赛C++提高组csp-s初赛真题及答案解析(阅读程序第1题)
c++·题解·真题·初赛·信奥赛·csp-s·提高组
逆境不可逃6 小时前
【春节篇】LeetCode 热题 100 之 238.除了自身以外数组的乘积
数据结构·算法·leetcode
散峰而望7 小时前
【算法竞赛】堆和 priority_queue
开发语言·数据结构·c++·算法·贪心算法·动态规划·推荐算法
alexwang2117 小时前
B2007 A + B 问题 题解
c++·算法·题解·洛谷