Almost Arithmetic Progression(思维,暴力 CodeForces 978D)

题目传送门

题意: 给你一个数组,你可以对数组中每一个元素进行+1,-1一次,或者不操作,问你是否能使这个数组变成等差数列,如果能请打印最少操作次数,如果不能打印-1。

1≤n≤100000

思路: 等差数列就有一个公差,我们可以枚举前两个数,一共9种操作,前两个数确定之后,这个数列公差就确定了,然后我们从3开始遍历,看每个数是否能通过操作或者不操作达到合法,维护最小值即可。

#include<bits/stdc++.h>
using namespace std;
const int N=2e5+5;
const int inf=0x7fffffff;
const int mod=1e9+7;
const int eps=1e-6;
typedef long long ll;
typedef unsigned long long ull;
#define ls p<<1
#define rs p<<1|1
#define fi first
#define se second
#define mp make_pair
#define pb push_back
#define pii pair<int,int>
#define int long long
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define endl '\n'
#define null NULL
int a[N],b[N];
signed main()
{
    IOS;
    int n;
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
        b[i]=a[i];
    }
    if(n<=2)
    {
        cout<<0<<endl;
        return 0;
    }
    int res=inf;
    for(int i=a[1]-1;i<=a[1]+1;i++)
    {
        for(int j=a[2]-1;j<=a[2]+1;j++)
        {
            int d=j-i;int cnt=0;
            if(i!=a[1])
                cnt++;
            if(j!=a[2])
                cnt++;
            for(int k=3;k<=n;k++)
            {
                int p=a[k-1];
                if(k==3)
                    p=j;
                if(abs(p+d-a[k])<=1)
                {
                    if(p+d-a[k]!=0)
                        cnt++;
                    a[k]=p+d;
                }
                else
                {
                    cnt=inf;
                    break;
                }
            }
            res=min(res,cnt);
            for(int k=3;k<=n;k++)
            {
                a[k]=b[k];
            }
        }
    }
    if(res==inf)
    {
        cout<<-1<<endl;
    }
    else
    {
        cout<<res<<endl;
    }
}
发布了93 篇原创文章 · 获赞 9 · 访问量 4203

猜你喜欢

转载自blog.csdn.net/Joker_He/article/details/105441165