1397B - Power Sequence 幂序列（思路+证明公式)

题意

Let’s call a list of positive integers a0,a1,…,an−1a0,a1,…,an−1 a power sequence if there is a positive integer cc, so that for every 0≤i≤n−10≤i≤n−1then ai=ciai=ci.

Given a list of nn positive integers a0,a1,…,an−1a0,a1,…,an−1, you are allowed to:

• Reorder the list (i.e. pick a permutation pp of {0,1,…,n−1}{0,1,…,n−1} and change aiai to apiapi), then
• Do the following operation any number of times: pick an index ii and change aiai to ai−1ai−1 or ai+1ai+1 (i.e. increment or decrement aiai by 11) with a cost of 11.

Find the minimum cost to transform a0,a1,…,an−1a0,a1,…,an−1 into a power sequence.

Input

The first line contains an integer nn (3≤n≤1053≤n≤105).

The second line contains nn integers a0,a1,…,an−1a0,a1,…,an−1 (1≤ai≤1091≤ai≤109).

Output

Print the minimum cost to transform a0,a1,…,an−1a0,a1,…,an−1 into a power sequence.

思路：

• 首先：升序重排一定是最优的，因为最终的Power Sequence 升序,因此我们升序排列一定最贴合最终序列。

对于$q,n$,假设$q=10,n=100$,最后一项就是$q=10^{100}$,非常之大，明显不符合题意，因此引发我们思考，去寻找$q$的取值范围。

下面简单证明一下$q$的取值范围

对于长度为n的序列$[a_1.....a_i]$，最起码的我们可以让其全部变为1满足Power Sequence，让其全变为的花费为$sum={\sum }_{i=1}^n(a_i-1)$

,现在来确定一下sum的取值范围，当$a_i=10^9$，$sum$为最大花费，即$sum<n\ast 10^9$。最终形成的Power Sequence，公比$q=c$，其第$n$项为最大值$a_n=q^{n-1}$，如果$a_n>sum$那么就毫无意义，因此需要满足$an<=sum$,即$q^{n-1}<=sum<n\ast 10^9$，化简一下就可以得到q的取值范围
$$1<=q<\sqrt[n-1]{n\ast 10^9}$$
我们枚举q难道会超时吗？其实不然，我们时间复杂度为 $O(n\ast q)=O(n\ast \sqrt[n-1]{n\ast 10^9})$，n=3的时候

我们发现n最小值为3，即
$$1<=q<\sqrt[n-1]{n\ast 10^9}<=\sqrt{3\ast 10^9}<10^5$$
n=4的时候$q<1e4$，很明显递减了(也可以对公式求导进行直接验证)，因此我们时间上限为
$$\sqrt{3\ast 10^9}<10^5$$
综上：不会超时

算法

枚举n和q算出即可，唯一注意的是
$$1<=q<\sqrt[n-1]{n\ast 10^9}<=\sqrt{3\ast 10^9}<10^5$$
后面式子c++会自动向下取整,因此等式修改为
$$1<q<=[\sqrt[n-1]{n\ast 10^9}]<=[\sqrt{3\ast 10^9}]<10^5$$

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e5+5;
int r[N];
int main() {
    int n;
    cin>>n;
    for(int i=0;i<n;i++){
        cin>>r[i];
    }
    sort(r,r+n);
    ll res=LLONG_MAX;
    ll up=pow(1e9*n,1.0/(n-1)); //最大值
    for(ll q=1;q<=up;q++){
        ll x=1,sum=0;
        for(int i=0;i<n;i++){
            sum+=abs(r[i]-x);
            x*=q;
        }
        res=min(res,sum);
    }
    cout<<res<<endl;
}