【初级中位数问题:货仓选址】利用中位数求最值
很简单,但是如果想不到就会比较麻烦。所以要熟练掌握思想精髓。
具体可以看题目poj 1732 (双重中位数+小技巧)
#include <cmath>
#include <iostream>
#include <cstdio>
#include <string>
#include <cstring>
#include <vector>
#include <algorithm>
#include <stack>
using namespace std;
typedef long long ll;
/*【士兵】poj 1723
有N个士兵,每个士兵站的位置用一个坐标(x,y)表示,
现在要让所有士兵站在同一个水平线,即y坐标相同且x坐标相邻,
每个士兵每次可以移动一个位置。求出最少的移动步数。 */
/*【分析】x方向和y方向分开讨论。
1.y方向:将所有坐标的y值从小到大排序,首尾两两配对,
最终所有士兵的y坐标相同,即最终的y值取中位数即可。
y_steps=|y[0]-y_mid|+|y[1]-y_mid|+...+|y[n-1]-y_mid|,y_mid=y[n/2]。
2.x方向:对所有x坐标从小到大排序,由于要移动步数最少,
所以【最终的x坐标相对位置】与【排序后的x坐标相对位置】相同。
设最终的x坐标起始位置为a,则:
x[0] -> a (->表示移动)
x[1] -> a+1
....
x[n-1] -> a+n-1
即 x[0] -> a
x[1]-1 -> a
x[2]-2 -> a
....
x[n-1]-n-1 ->a ... 将所有x[i]-i移动到相同位置。
于是重新计算x[i]=x[i]-i,然后重新进行排序。
x_mid=x[n/2],x_steps=|x[0]-x_mid|+|x[1]-x_mid|+...+|x[n-1]-x_mid|。 */
int cmp(const void *a, const void *b){
//↑↑↑不管传入什么类型,cmp函数内部将该指针转换为int类型
return *(int*)a-*(int*)b;
} //将a所指向地址的数据转换为int型,b同理,将两数相减,返回结果。
#define maxn 10020
int x[maxn],y[maxn],n;
int main(){
while(scanf("%d",&n)!=EOF){
int ans=0;
for(int i=0;i<n;i++)
scanf("%d%d",&x[i],&y[i]);
sort(y,y+n); //处理y
int mid=n/2; //找到中位数(从0开始,奇数情况正好达到中间值)
//(如果总数是偶数,选择其中的任意一个,答案相同)
for(int i=0;i<mid;i++) ans+=y[mid]-y[i];
for(int i=mid;i<n;i++) ans+=y[i]-y[mid];
sort(x,x+n); //处理x
for(int i=0;i<n;i++) x[i]-=i; //相当于x[n-1]-n-1移到a
sort(x,x+n); //重新排序 //↓↓↓仍设定在中间的mid位置处
for(int i=0;i<mid;i++) ans+=x[mid]-x[i];
for(int i=mid;i<n;i++) ans+=x[i]-x[mid];
cout<<ans<<endl;
}
return 0;
}