[codechef]UASEQ

题目大意

给定一个序列，修改至多k次，变成等差数列。
最小化{首项，公差}的字典序。
k<=min(n-2,10)。

做法

因为k<=n-2，一定有两个位置不会被修改。
随机两个位置，以它们作为不修改项。
做不会超时次。
n很小时，最优解一定能被枚举到。
n很大时，不被修改的位置有n-k个，正确率极高。

#include<cstdio>
#include<algorithm>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
typedef long long ll;
const int maxn=100000+10;
ll a[maxn],b[maxn],c[maxn];
int i,j,k,l,n,m,ca;
ll t,d;
int main(){
    scanf("%d%d",&n,&m);
    b[1]=10000000000000000;
    fo(i,1,n) scanf("%d",&a[i]);
    ca=10000000/n;
    while (ca--){
        i=rand()*rand()%n+1;j=rand()*rand()%n+1;
        if (i>j) swap(i,j);
        if (i==j) continue;
        d=(a[j]-a[i])/(j-i);
        c[1]=a[i]-(ll)(i-1)*d;
        if (c[1]>b[1]||c[1]==b[1]&&d>=b[2]-b[1]) continue;
        fo(k,2,n) c[k]=c[k-1]+d;
        t=0;
        fo(k,1,n) t+=(a[k]!=c[k]);
        if (t<=m){
            fo(k,1,n) b[k]=c[k];
        }
    }
    fo(i,1,n) printf("%lld ",b[i]);
}