【C++ 真题】P1678 烦恼的高考志愿

P1678 烦恼的高考志愿

题目背景

计算机竞赛小组的神牛 V 神终于结束了高考，然而作为班长的他还不能闲下来，班主任老 t 给了他一个艰巨的任务：帮同学找出最合理的大学填报方案。可是 v 神太忙了，身后还有一群小姑娘等着和他约会，于是他想到了同为计算机竞赛小组的你，请你帮他完成这个艰巨的任务。

题目描述

现有 $m$ 所学校，每所学校预计分数线是 $a_i$。有 $n$ 位学生，估分分别为 $b_i$。

根据 $n$ 位学生的估分情况，分别给每位学生推荐一所学校，要求学校的预计分数线和学生的估分相差最小（可高可低，毕竟是估分嘛），这个最小值为不满意度。求所有学生不满意度和的最小值。

输入格式

第一行读入两个整数 $m,n$。$m$ 表示学校数，$n$ 表示学生数。

第二行共有 $m$ 个数，表示 $m$ 个学校的预计录取分数。第三行有 $n$ 个数，表示 $n$ 个学生的估分成绩。

输出格式

输出一行，为最小的不满度之和。

输入输出样例 #1

输入 #1

4 3
513 598 567 689
500 600 550

输出 #1

32

说明/提示

数据范围：

对于 $30\%$ 的数据，$1\le n,m\le 1000$，估分和录取线 $\le 10000$；

对于 $100\%$ 的数据，$1\le n,m\le 100000$，估分和录取线 $\le 1000000$ 且均为非负整数。

题解

注意：这里sum要开 long long 数据类型，不然洛谷最后一项测试会卡住。

#include "bits/stdc++.h"
using namespace std;
const int N = 1e6+7;
int n, m, a, b;
int g1[N], g2[N];
long long sum = 0;
int main() {
	cin>>m>>n;
	for(int i=1;i<=m;++i){
		cin>>g1[i];
	}
	for(int i=1;i<=n;++i){
		cin>>g2[i];
	}
	sort(g1+1,g1+m+1);
	for(int i=1;i<=n;++i){
		int l=0, r=m+1;
		while(l<r){
			int mid = l+(r-l)/2; 
			if(g1[mid]>=g2[i]){
				r = mid;
			}else if(g1[mid]<=g2[i]){
				l = mid+1;
			}
		}
		if(g2[i]<=g1[1]){
			sum += g1[1] - g2[i];
		}else if(g2[i]>=g1[m]){
			sum += abs(g1[m] - g2[i]);
		}else{
			sum += min(abs(g2[i]-g1[l-1]), abs(g2[i]-g1[l]));
		}
	}
	cout<<sum<<endl;
	return 0;
}