CF1185D Extra Element(细节)

传送门

思路:

首先要明确的问题,怎么判断该序列是等差序列:
从小到大排序后,所有的邻项的差值都相等,也就是差值出现了\(n-2\)次。
所以说,可以枚举删除哪个数,然后\(check\)是否合法。
为什么可行呢?
因为删除该数的时候影响的只是他的相邻项,可以预处理出邻项的差值,然后每次\(O(1)check\)
注意边界问题就好了。

没注意细节wa了两发

代码:

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <cmath>
#include<map>
#include<set>
#include<vector>
#include<queue>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<ll, ll>PLL;
typedef pair<int, int>PII;
typedef pair<double, double>PDD;
#define I_int ll
inline ll read()
{
    ll x = 0, f = 1;
    char ch = getchar();
    while(ch < '0' || ch > '9')
    {
        if(ch == '-')f = -1;
        ch = getchar();
    }
    while(ch >= '0' && ch <= '9')
    {
        x = x * 10 + ch - '0';
        ch = getchar();
    }
    return x * f;
}
#define read read()
#define closeSync ios::sync_with_stdio(0);cin.tie(0);cout.tie(0)
#define multiCase int T;cin>>T;for(int t=1;t<=T;t++)
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define repp(i,a,b) for(int i=(a);i<(b);i++)
#define per(i,a,b) for(int i=(a);i>=(b);i--)
#define perr(i,a,b) for(int i=(a);i>(b);i--)
ll ksm(ll a, ll b, ll p)
{
    ll res = 1;
    while(b)
    {
        if(b & 1)res = res * a % p;
        a = a * a % p;
        b >>= 1;
    }
    return res;
}
const int inf = 0x3f3f3f3f;
#define PI acos(-1)
const int maxn=5e5+100;

ll n;
struct node{
    ll val,id;
}a[maxn];

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

map<ll,ll>mp;

int main()
{
    n=read;
    rep(i,1,n){
        a[i].val=read;
        a[i].id=i;
    }
    if(n==2){
        puts("1");return 0;
    }
    sort(a+1,a+1+n,cmp);
    for(int i=2;i<=n;i++) mp[a[i].val-a[i-1].val]++;
    for(int i=1;i<=n;i++){///删哪个
        if(i==1){
            mp[a[2].val-a[1].val]--;
            if(mp[a[3].val-a[2].val]==n-2){
                cout<<a[1].id<<endl;
                return 0;
            }
            mp[a[2].val-a[1].val]++;
        }
        else if(i==n){
            mp[a[n].val-a[n-1].val]--;
            if(mp[a[n-1].val-a[n-2].val]==n-2){
                cout<<a[n].id<<endl;
                return 0;
            }
            mp[a[n].val-a[n-1].val]++;
        }
        else{
            mp[a[i].val-a[i-1].val]--;mp[a[i+1].val-a[i].val]--;
            mp[a[i+1].val-a[i-1].val]++;
            if(mp[a[i+1].val-a[i-1].val]==n-2){
                cout<<a[i].id<<endl;
                return 0;
            }
            mp[a[i].val-a[i-1].val]++;mp[a[i+1].val-a[i].val]++;
            mp[a[i+1].val-a[i-1].val]--;
        }
    }
    puts("-1");
    return 0;
}

/*

**/


posted @ 2021-06-03 21:12  OvO1  阅读(55)  评论(0编辑  收藏  举报