agc 027 B - Garbage Collector

B - Garbage Collector

https://agc027.contest.atcoder.jp/tasks/agc027_b

题意：

　　x坐标轴上n个垃圾，有一个机器人在从原点，要清扫垃圾。原点有一个垃圾桶。机器人可以在x轴上左右移动，当移动到某个垃圾的位置上时，可以选择花费 X 点能量将它捡起来（也可以视而不捡）。机器人如果到达垃圾桶，则可以将它携带的垃圾花费 X 点能量倒出。机器人如果携带着 K 件垃圾移动一个单位距离，则需要消耗 (K+1)^2 点能量。问将所有垃圾全部弄到垃圾桶里面去所需消耗的最小能量。

分析：

　　贪心 + 推性质。

　　如果机器人一次将多个垃圾一起捡走，那么一定是从走到最后面，再回来。如果机器人第i个垃圾回来，花费为$pos_i \times (1 + 1) ^ 2 + pos_i = 5 pos_i$，如果在i前面再捡一个垃圾，新增加的花费为：$pos_j \times ((1 + 2) ^ 2 - (1 + 1) ^ 2) = 5 pos_j$ 同样再增加一个：$pos_k \times ((1 + 3) ^ 2 - (1 + 2) ^ 2) = 7pos_k$。所以有公式：

　　$F(i)=\left\{\begin{matrix} 5pos\ \ (i=1)\\ (2i+1)pos\ \ (i>1) \end{matrix}\right.$

　　表示第i个捡的垃圾的花费。

　　因为系数一样，所以每次捡多少应该都是一样的，枚举每次捡了多少。然后贪心的思路，让最远的点乘以系数最小的，就是让最远的k个点都是第一次拿，次远的k个点都是第二次拿...

代码：

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<iostream>
#include<cctype>
#include<set>
#include<vector>
#include<queue>
#include<map>
#define fi(s) freopen(s,"r",stdin);
#define fo(s) freopen(s,"w",stdout);
using namespace std;
typedef long long LL;

inline LL read() {
    LL x=0,f=1;char ch=getchar();for(;!isdigit(ch);ch=getchar())if(ch=='-')f=-1;
    for(;isdigit(ch);ch=getchar())x=x*10+ch-'0';return x*f;
}

LL a[200005];

int main() {
    int n = read();LL x = read();
    for (int i=1; i<=n; ++i) a[i] = read() + a[i - 1];
    LL ans = 1e18;
    for (int k=1; k<=n; ++k) {
        LL sum = 0, now = 3;
        for (int i=n; i>=1; i-=k) {
            sum += (a[i] - a[max(0, i - k)]) * max(now, 5ll), now += 2;
            if (sum >= ans) break;
        }
        ans = min(ans, sum + 1ll * (k + n) * x);
    }
    cout << ans;
    return 0;
}