题解: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;
}
相关推荐
历程里程碑20 小时前
53 多路转接select
linux·开发语言·数据结构·数据库·c++·sql·排序算法
谙弆悕博士1 天前
【附C源码】从零实现C语言堆数据结构:原理、实现与应用
c语言·数据结构·算法··数据结构与算法
smj2302_796826522 天前
解决leetcode第3934题最短唯一子数组
数据结构·python·算法·leetcode
iiiiyu2 天前
面向对象和集合编程题
java·开发语言·前端·数据结构·算法·编程语言
变量未定义~2 天前
最长回文子串
数据结构·算法
代码中介商2 天前
AVL树:自平衡二叉搜索树的奥秘
数据结构
玛卡巴卡ldf2 天前
【LeetCode 手撕算法】(多维动态规划)不同路径、最小路径和、最长回文子串、最长公共子序列、编辑距离
java·数据结构·算法·leetcode·动态规划·力扣
被AI抢饭碗的人2 天前
算法:数据结构
数据结构·算法
淞綰2 天前
c语言的练习-字符串的练习-寻找最长连续字符以及出现次数
c语言·数据结构·学习·算法·c语言的练习
qq_296553272 天前
[特殊字符] 搜索插入位置:从O(n)到O(log n)的优雅进化
数据结构·算法·面试·分类·柔性数组