CF-1185D-Extra Element(二分,暴力)

题目链接:https://codeforces.com/problemset/problem/1185/D

题目大意:给出一个数组A,判断里面的数删掉一个是否能够组成一个等差数列。如果可以,输出删除的那个数的位置。否则输出-1.

思路:将数组排一下序,然后将所有的公差都放入map中。然后枚举每个元素,判断是否可以删除。特判一下第一个和最后一个。

ACCode:

#include<stdlib.h>
#include<string.h>
#include<stdio.h>
#include<time.h>
#include<math.h>
// srand((unsigned)time(NULL));rand();
      
#include<map>//unordered_map
#include<set>//multiset
#include<deque>
#include<queue>
#include<stack>
#include<bitset>
#include<string>
#include<fstream>
#include<iostream>
#include<algorithm>
 
#define ll long long
#define PII pair<int,int>
#define PLL pair<ll,ll>
#define clean(a,b) memset(a,b,sizeof(a))
using namespace std;
      
const int MAXN=2e5+10;
//const int MAXM=10;
const int INF32=0x3f3f3f3f;
const ll INF64=0x3f3f3f3f3f3f3f3f;
const ll MOD=1e9+7;
const double PI=acos(-1.0);
const double EPS=1.0e-8;
//unsigned register
// ios::sync_with_stdio(false)

int A[MAXN],B[MAXN];
map<int,int> Vis;
int n;

int PosX(int l,int r,int Key,int mid=0){
	while(l<=r){
		mid=(l+r)>>1;
		if(A[mid]==Key) return mid;
		else if(A[mid]>Key) r=mid-1;
		else l=mid+1;
	}
}
int main(){
	while(~scanf("%d",&n)){
		Vis.clear();
		for(int i=1;i<=n;++i) scanf("%d",&A[i]);
		for(int i=1;i<=n;++i) B[i]=A[i];
		if(n==2||n==3){
			printf("1\n");continue ;
		}
		sort(A+1,A+1+n);
		for(int i=1;i<=n-1;++i) Vis[A[i+1]-A[i]]++;
		int flag=0;
		for(int i=1;i<=n;++i){
			int pos=PosX(1,n,B[i]);
			if(pos==1){//特判起点 
				int d=A[2]-A[1];
				Vis[d]--;
				int cnt=0;
				for(auto it:Vis){
					if(it.second>0) cnt++;
					if(cnt>1) break;
				} 
				if(cnt==1){
					flag=i;break;
				}
				else Vis[d]++;
				continue ;
			}
			if(pos==n){//特判终点 
				int d=A[n]-A[n-1];
				Vis[d]--;
				int cnt=0;
				for(auto it:Vis){
					if(it.second>0) cnt++;
					if(cnt>1) break;
				}
				if(cnt==1){
					flag=i;break;
				}
				else Vis[d]++;
				continue ;
			}
			int d1=A[pos]-A[pos-1],d2=A[pos+1]-A[pos],d3=A[pos+1]-A[pos-1];
			Vis[d1]--;Vis[d2]--;Vis[d3]++;
			int cnt=0;
			for(auto it:Vis){
				if(it.second>0) cnt++;
				if(cnt>1) break;
			}
			if(cnt==1){
				flag=i;break;
			}
			Vis[d1]++;Vis[d2]++;Vis[d3]--;
		}
		if(flag>0) printf("%d\n",flag);
		else printf("-1\n");
	}
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值