(hzau)华中农业大学第四届程序设计大赛网络同步赛 G: Array C

题目链接：http://acm.hzau.edu.cn/problem.php?id=18

题意是给你两个长度为n的数组，a数组相当于1到n的物品的数量，b数组相当于物品价值，而真正的价值表示是b[i]*k*k(k表示取的数量)，给你m表示最少取m个物品。然后让你求取m个的最小的价值和。

一个物品的价值是b[i] , 两个物品的价值是b[i] * 3 + b[i] ... 

所以每个物品的价值是b[i] , b[i] * 3 , b[i] * 5 , b[i] * 7 ...

每个物品数量a[i]不超过100，所以总共不超过1e5个。所有的物品价值存到数组中，sort一下取前m个即可。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long LL;
LL cost[100005] , a[1005] , b[1005];
int main()
{
    int n , m;
    while(~scanf("%d %d" , &n , &m)) {
        for(int i = 1 ; i <= n ; ++i) {
            scanf("%lld" , a + i); 
            a[i] = min(a[i] , (LL)m);
        }
        int cnt = 0;
        for(int i = 1 ; i <= n ; ++i) {
            scanf("%lld" , b + i);
            for(int j = 1 ; j <= a[i] ; ++j)
                cost[cnt++] = (LL)(2 * j - 1) * b[i];
        }
        sort(cost , cost + cnt);
        LL sum = 0;
        for(int i = 0 ; i < m ; ++i) {
            sum += cost[i];
        }
        printf("%lld\n" , sum);
    }
}