Google KickStart 2020 Round G, Combination Lock，二分

题目链接

题意：

给$N$个数，每个数的范围为$[1,W]$，通过以下两种操作将所有的数都变成一样的，问最小操作数是多少。

1. 操作一个数$+1$，即$a[i]=(a[i]+1>W?1:a[i]+1)$
2. 操作一个数$-1$，即$a[i]=(a[i]-1<1?W:a[i]-1)$

分析：

这道题里对每个数$+1,-1$是可以溢出的，若没有溢出的话，很明显求个中位数即可。但是这里有溢出的操作的话，就需要比较哪种操作更好了。
假设最终数组里数都变成了$val$，通过观察可以发现$val$一定是原数组$a[1...n]$中的其中一个，为什么？下面我们来证明这个结论：
因为$val!=a[1...n]$，假设此时数组$a$从小到大有序，那么有小于$val$的最大下标$i$，有大于$val$的最小下标$j$，假设当前有$x$个数已经变成了$a[i]$，那么有$n-x$个数变成了$a[j]$，那么还需操作$x\ast (val-a[i])+(n-x)\ast (a[j]-val)$次，才能让所有的数都变成$val$。
考虑下面两种情况，无论哪一种，都可以得到更少的操作次数:

• $x\le n-x$. 如果将所有的数都变成$a[j]$，那么还需操作$x\ast (a[j]-a[i])$次，等同于$x\ast (val-a[i])+x\ast (a[j]-val)$，因为$x\le n-x$，显然操作次数更少；
• $x\ge n-x$. 如果将所有的数都变成$a[i]$，那么还需操作$(n-x)\ast (a[j]-a[i])$次，等同于$(n-x)\ast (val-a[i])+(n-x)\ast (a[j]-val)$，显然操作次数更少。

所以最终一定是变成$a[1...n]$中的某个初始值。

那么首先对数组$a$排序，然后依次遍历每个值，现在让我们来考虑对于每一个$a[i]$，应该如何计算操作次数。

• $j<i$时，$cnt\_op=min(a[i]-a[j],a[j]+W-a[i])$
• $j>i$时，$cnt\_op=min(a[j]-a[i],W-a[j]+a[i])$

对于$j<i$这一段来说，$a[i]-a[j]$是随着$j$的增大递减的，$a[j]+W-a[i]$是递增的，也就是说这两个值之间存在某个交点$k$，在$[1,k-1]$这一段中最小值是$a[j]+W-a[i]$，在$[k,i]$这一段中最小值是$a[i]-a[j]$，所以用二分求出$k$的位置(第一个位置使得$a[i]-a[k]\le a[k]+W-a[i]$)，$j<i$这一段的操作数就是$(k-1)\ast (W-a[i])-{\sum }_{j=1}^{k-1}a[j]+(i-k+1)\ast a[i]-{\sum }_{j=k}^{i}a[j]$.
对于$j>i$这一段来说，$a[j]-a[i]$随着$j$的增大递增，$W-a[j]+a[i]$递减，同样用二分找到$k$的位置使得$a[k]-a[i]\le W-a[j]+a[i]$成立的最后一个位置，然后需要的操作数就是${\sum }_{j=i+1}^{k}a[j]-(k-i)\ast a[i]+(n-k)\ast (w+a[i])-{\sum }_{j=k+1}^{n}a[j]$.
将两段加起来即为总的操作数。

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;

int main() {
	int t, tt = 0;
	cin >> t;
	while (t--) {
		int n, w;
		cin >> n >> w;
		vector<ll> a(n+1), pre(n+1);
		for (int i = 1; i <= n; i++) {
			cin >> a[i];
		}
		
		sort(a.begin()+1, a.end());
		for (int i = 1; i <= n; i++) {
			pre[i] = pre[i-1] + a[i];
		}
		
		ll res = LONG_LONG_MAX;
		for (int i = 1; i <= n; i++) {
			ll sum = 0;
			int l = 1, r = i, mid;
			// 这里取右端点，原因是可能在i=1的位置就有rvalue <= lvalue
			while (l < r) {
				mid = (l + r) / 2;
				ll lvalue = w - a[i] + a[mid], rvalue = a[i] - a[mid];
				if (lvalue == rvalue) {
					l = r = mid;
				}
				if (lvalue < rvalue) {
					l = mid+1;
				} else {
					r = mid;
				}
			}
			
			sum += ((l-1) * (w - a[i]) + pre[l-1]) + 
					((i-l+1) * a[i] - (pre[i] - pre[l-1]));
			
			l = i, r = n;
			// 这里取左端点，原因可能是在i=n的位置才有rvalue <= lvalue
			while (l < r) {
				mid = (l + r + 1) / 2;
				ll lvalue = a[mid] - a[i], rvalue = w - a[mid] + a[i];
				if (lvalue == rvalue) {
					l = r = mid;
				}
				if (lvalue < rvalue) {
					l = mid;
				} else {
					r = mid - 1;
				}
			}
			
			sum += ((pre[l] - pre[i]) - (l-i) * a[i]) + 
					((n - l) * (w + a[i]) - (pre[n] - pre[l]));
			
			res = min(res, sum);
		}
		cout << "Case #" << ++tt << ": ";
		cout << res << endl;
	}
}