Codeforce:Round 568 div2 D

博客围绕CodeForces相关模拟展开,提到模拟时删除元素有三种方式,即删第一个、删第二个或删中间的,且第一个和第二个删除需确定等差值。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

模拟:

要么删第一个,要么删第二个,要么删之后中间的
第一个和第二个需要确定等差值

#include<bits/stdc++.h>

using namespace std;

#define maxn 200005

int n;
struct Node{
    int id;
    long long v;
};
Node num[maxn];

bool cmp(const Node& x,const Node& y){
    if( x.v == y.v){
        if( x.id < y.id ) return true;
        else return false;
    }
    if(x.v < y.v) return true;
    return false;
}

int main(){
    cin>>n;
    for(int i=0;i<n;i++){
        cin>>num[i].v;
        num[i].id = i+1;
    }
    sort(num,num+n,cmp);
    // delete 1
    long long key = num[2].v - num[1].v;
    bool ans = true;
    for(int i=3;i<n;i++){
        if(num[i].v - num[i-1].v != key){
            ans = false;
            break;
        }
    }
    if(ans){
        cout<<num[0].id<<endl;
        return 0;
    }

    // delete second
    key = num[2].v - num[0].v;
    ans = true;
    for(int i=3;i<n;i++){
        if(num[i].v - num[i-1].v != key){
            ans = false;
            break;
        }
    }
    if(ans){
        cout<<num[1].id<<endl;
        return 0;
    }

    // delete mid
    key = num[1].v - num[0].v;
    ans = true;
    for(int i=2;i<n;i++){
        if(num[i].v - num[i-1].v != key){
            // 删除 i 
            if(i+1<n){
                if(num[i+1].v - num[i-1].v == key){
                    for(int j=i+2;j<n;j++){
                        if(num[j].v - num[j-1].v != key){
                            cout<<-1<<endl;
                            return 0;
                        }
                    }
                    cout<<num[i].id<<endl;
                    return 0;
                }else{
                    cout<<-1<<endl;
                    return 0;
                }
            }else{
                cout<<num[i].id<<endl;
                return 0;
            }
        }
    }
    cout<<num[0].id<<endl;
    return 0;  
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值