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=,f=;char ch=getchar();for(;!isdigit(ch);ch=getchar())if(ch=='-')f=-;
     for(;isdigit(ch);ch=getchar())x=x*+ch-'';return x*f;
 }

 LL a[];

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