1112: [POI2008]砖块Klo

原创 2016年08月30日 20:39:45

1112: [POI2008]砖块Klo

Time Limit: 10 Sec  Memory Limit: 162 MB
Submit: 1664  Solved: 583
[Submit][Status][Discuss]

Description

N柱砖,希望有连续K柱的高度是一样的. 你可以选择以下两个动作 1:从某柱砖的顶端拿一块砖出来,丢掉不要了. 2:从仓库中拿出一块砖,放到另一柱.仓库无限大. 现在希望用最小次数的动作完成任务.

Input

第一行给出N,K. (1 ≤ k ≤ n ≤ 100000), 下面N行,每行代表这柱砖的高度.0 ≤ hi ≤ 1000000

Output

最小的动作次数

Sample Input

5 3
3
9
2
3
1

Sample Output

2

HINT

原题还要求输出结束状态时,每柱砖的高度.本题略去.

Source

[Submit][Status][Discuss]

提问:给定一维坐标系上的k个点,找一个点,使得它到其它点的距离和最小
找中位数啊。。。。
维护一个treap就行
#include<iostream>
#include<cstdio>
#include<queue>
#include<vector>
#include<bitset>
#include<algorithm>
#include<cstring>
#include<map>
#include<stack>
#include<set>
#include<cmath>
#include<ext/pb_ds/priority_queue.hpp>
using namespace std;

const int maxn = 1E5 + 10;
typedef long long LL;

int n,k,cnt,Root,a[maxn],ch[maxn][2],key[maxn],va[maxn],siz[maxn];
LL Ans = 1E18,sum[maxn];

LL Sum(int x,int tot)
{
	if (!tot) return 0;
	if (siz[x] <= tot) return sum[x];
	int Left = siz[ch[x][0]] + 1;
	if (Left > tot) return Sum(ch[x][0],tot);
	else return sum[ch[x][0]] + 1LL*va[x] + Sum(ch[x][1],tot - Left);
}

int Rank(int x,int tot)
{
	int Left = siz[ch[x][0]] + 1;
	if (Left == tot) return va[x];
	else if (Left > tot) return Rank(ch[x][0],tot);
	else return Rank(ch[x][1],tot - Left);
}

void Solve(LL Now,int mid1,int mid2)
{
	LL z = mid1,sum = Sum(Root,mid1);
	LL tot = Now*z - sum;
	sum = Sum(Root,k) - Sum(Root,mid1);
	tot += (sum - Now*z);
	Ans = min(Ans,tot);
}

void maintain(int x)
{
	sum[x] = va[x];
	siz[x] = 1;
	for (int i = 0; i < 2; i++)
		if (ch[x][i]) {
			siz[x] += siz[ch[x][i]];
			sum[x] += sum[ch[x][i]];
		}
}

void rotate(int &x,int d)
{
	int y = ch[x][d];
	ch[x][d] = ch[y][d^1];
	maintain(x);
	ch[y][d^1] = x;
	x = y; maintain(x);
}

int cmp(int x,int w)
{
	if (va[x] == w) return -1;
	if (va[x] < w) return 1;
	return 0;
}

int New(int w)
{
	int ret = ++cnt;
	ch[ret][0] = ch[ret][1] = 0;
	key[ret] = rand();
	va[ret] = sum[ret] = w;
	siz[ret] = 1;
	return ret;
}

void Insert(int &x,int w)
{
	if (!x) {
		x = New(w);
		return;
	}
	int d = cmp(x,w);
	d = max(d,0);
	Insert(ch[x][d],w);
	maintain(x);
	if (key[ch[x][d]] > key[x]) rotate(x,d);
}

void Remove(int &x,int w)
{
	int d = cmp(x,w);
	if (d == -1) {
		if (!ch[x][0] && !ch[x][1]) x = 0;
		else if (!ch[x][0] || !ch[x][1]) 
			x = ch[x][0]?ch[x][0]:ch[x][1];
		else {
			int d2 = key[ch[x][0]] > key[ch[x][1]]?0:1;
			rotate(x,d2);
			Remove(ch[x][d2^1],w);
			maintain(x);
			if (key[ch[x][d2^1]] > key[x])
				rotate(x,d2^1);
		}
		return;
	}
	Remove(ch[x][d],w);
	maintain(x);
	if (key[ch[x][d]] > key[x])
		rotate(x,d);
}

int main()
{
	#ifdef DMC
		freopen("DMC.txt","r",stdin);
	#endif
	
	cin >> n >> k;
	if (k == 1) {
		cout << 0;
		return 0;
	}
	for (int i = 1; i <= n; i++) 
		scanf("%d",&a[i]);
	for (int i = 1; i <= k; i++)
		Insert(Root,a[i]);
	for (int i = k; i <= n; i++) {
		if (i > k) {
			Remove(Root,a[i-k]);
			Insert(Root,a[i]);
		}
		if (k&1) {
			int mid = (k + 1) >> 1;
			LL Now = Rank(Root,mid),tot = 0;
			LL z = mid-1,sum = Sum(Root,mid-1);
			tot += (z*Now - sum);
			sum = Sum(Root,k) - Sum(Root,mid);
			tot += (sum - z*Now);
			Ans = min(Ans,tot);
		}
		else {
			int mid1 = k>>1,mid2 = mid1+1;
			int Now1 = Rank(Root,mid1);
			int Now2 = Rank(Root,mid2);
			int Now = (Now1 + Now2) >> 1;
			Solve(Now,mid1,mid2);
			if (Now1 != Now2) 
				Solve(Now + 1,mid1,mid2);
		}
	}
	cout << Ans;
	return 0;
}

版权声明:这个人很懒什么都没有留下

相关文章推荐

【BZOJ】1112 [POI2008]砖块Klo 平衡树

题目传送门 今天一晚上就做了一道题,好颓废啊…… 这题其实还挺水的,只不过本蒟蒻一直都不太想的进去,所以一直WA,一直懵逼。 其实这题的题意是在n个数中连续取k个,把这k个数都改成一个相同的数m,使得...
  • lyfsb
  • lyfsb
  • 2017年07月04日 21:27
  • 110

Bzoj1112:[POI2008]砖块Klo:splay

题目链接:1112:[POI2008]砖块Klo 原题目是要输出最终高度的,代码中有体现 显然一段区间都变成某个数且代价最小,那么就变成中位数 所以我们维护中位数即可,上splay 一不小心把树建反成...

【BZOJ】【P1112】【POI2008】【砖块Klo】【题解】【map】

传送门:http://www.lydsy.com/JudgeOnline/problem.php?id=1112 做法非常好想

【BZOJ 1112】 [POI2008]砖块Klo

splay维护中位数~

BZOJ 1112 [POI2008]砖块Klo Treap

BZOJ 1112 [POI2008]砖块Klo Treap
  • wzq_QwQ
  • wzq_QwQ
  • 2015年09月14日 20:17
  • 927

BZOJ1112: [POI2008]砖块Klo Treap

BZOJ 1112: [POI2008]砖块Klo Time Limit: 10 Sec  Memory Limit: 162 MB Submit: 1701  Solved: 597 ...
  • Oakley_
  • Oakley_
  • 2016年10月21日 11:26
  • 215

【POI2008】【BZOJ1112】砖块Klo

DescriptionN柱砖,希望有连续K柱的高度是一样的. 你可以选择以下两个动作 1:从某柱砖的顶端拿一块砖出来,丢掉不要了. 2:从仓库中拿出一块砖,放到另一柱.仓库无限大. 现在希望用最小次数...

[POI2008]砖块Klo

那么问题就在于如何找任意区间的中位数 划分树和平衡树都可以解决 所以是水题一道…… 下面描述我的悲惨经历…… 上周周末就看到了这道题 觉得应该可以用划分树水过 屁颠屁颠写了个划分树,写到一...
  • OrpineX
  • OrpineX
  • 2012年01月10日 19:42
  • 1154

BZOJ 1112 POI2008 砖块

POI , 树状数组
  • Fuxey
  • Fuxey
  • 2016年02月10日 15:49
  • 606

TYVJ P3568 砖块Klo

 时间: 1000ms / 空间: 165536KiB             描述 N柱砖,希望有连续K柱的高度是一样的. 你可以选择以下两个动作 1:从某柱砖的顶端拿一块...
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:1112: [POI2008]砖块Klo
举报原因:
原因补充:

(最多只允许输入30个字)