CF - D. Extra Elemen（multiset）

https://codeforces.com/contest/1185/problem/D

题意
给定 n 个数，判断是否可以删除一个位置，使得其余所有位置重新排序后形成等差数列？输出任意位置。

$2\le n\le 2\cdot 10^5,-10^9\le b_i\le 10^9$

思路
一开始想的很简单，遍历删掉的位置，然后判断剩余所有数的公差是否相同，然后又想，判断所有数是否相同要 O(n)，整体复杂度就是 $n^2$，然后就没写。。
然后就想的模拟，把所有公差都找出来，然后把出现次数最多的那个作为公差，然后判断不同的那个位置删掉之后形成的数能否和这个公差相同。还是挺麻烦的，写了几十行不对。
其实，第一次想的思路是对的，但是判断所有数是否相同不是 O(n)，而可以 O(logN)，把所有数存到 multiset 里，然后判断最大值和最小值是否相同。

具体实现：
把所有公差存在 multiset 里，遍历所有删除的位置，把在 set 里的公差删掉，然后加进去新的公差，然后通过判断最大值和最小值是否相同来看是否所有公差都相同了。
（注意判断过之后恢复原样再到下一位置）

Code

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

#define Ios ios::sync_with_stdio(false),cin.tie(0)

const int N = 200010, mod = 1e9+7;
int T, n, m;
struct node{
	int id, val;
}a[N];

bool cmp(node a, node b){
	return a.val < b.val;
}

signed main(){
	Ios;
	cin >> n;
	for(int i=1;i<=n;i++) cin >> a[i].val, a[i].id = i;
	
	if(n == 2){
		cout << 1;
		return 0;
	}
	
	sort(a+1, a+n+1, cmp);
	
	multiset<int> st;
	for(int i=2;i<=n;i++) st.insert(a[i].val - a[i-1].val);
	
	int flag = 0;
	for(int i=1;i<=n;i++)
	{
		if(i == 1)
		{
			st.erase(st.find(a[2].val - a[1].val));
			int mina = *st.begin(), maxa = *st.rbegin();
			if(mina == maxa) flag = 1;
			st.insert(a[2].val - a[1].val);
		}
		else if(i == n)
		{
			st.erase(st.find(a[n].val - a[n-1].val));
			int mina = *st.begin(), maxa = *st.rbegin();
			if(mina == maxa) flag = 1;
			st.insert(a[n].val - a[n-1].val);
		}
		else
		{
			st.erase(st.find(a[i+1].val - a[i].val));
			st.erase(st.find(a[i].val - a[i-1].val));
			st.insert(a[i+1].val - a[i-1].val);
			int mina = *st.begin(), maxa = *st.rbegin();
			if(mina == maxa) flag = 1;
			st.erase(st.find(a[i+1].val - a[i-1].val));
			st.insert(a[i+1].val - a[i].val);
			st.insert(a[i].val - a[i-1].val);
		}
		
		if(flag){
			cout << a[i].id;
			return 0;
		}
	}
	cout << -1;
	
	return 0;
}

经验
利用 multiset 快速求所有数中的最大值和最小值，还能实时修改，很好用！
要想起来用！！