D. Extra Element
题意:给一串数字,让删一个数字使使这个串排序后可以形成等差数列,可以的话输出这个数字开始时的下标,否则输出-1
思路:先用pair(int,int)使数字的值和位置形成映射关系,然后按first排序,之后顺着一对一对看它们之间的差值是否改变,如果变了就存在错误的数字,就分别检查看这两个是否可删除后使序列正常。
#include<bits/stdc++.h>
using namespace std;
const int maxn=2e5+10;
vector<pair<int,int>> v;
int flag=0;
void check(int pos)
{
vector<pair<int,int>> a;
for(int i=0; i<v.size(); i++)
{
if(pos!=i)
{
a.push_back(v[i]);
}
}
if(a.size()>2)
{
int cha=a[1].first-a[0].first;
for(int i=0; i<a.size()-1; i++)
{
if((a[i+1].first-a[i].first)!=cha)
return;
}
}
printf("%d", v[pos].second);
flag=1;
return;
}
int main()
{
int n;
scanf("%d",&n);
for(int i=1; i<=n; i++)
{
int t;
scanf("%d",&t);
v.push_back(make_pair(t,i));
}
int cha=-0x3f3f3f3f;
int cnt=0;
sort(v.begin(),v.end());
for(int i=1; i<=n-1; i++)
{
if((v[i].first-v[i-1].first)!=cha)
{
cnt++;
cha=v[i].first-v[i-1].first;
if(cnt>2)
break;
// printf("%d %d\n",i,i-1);
check(i);
if(flag==1) return 0;
check(i-1);
if(flag==1) return 0;
}
}
printf("-1");
}