秒客网

Problem Description

There are n kinds of goods in the company, with each of them has a inventory of cnt_i and direct unit benefit val_i. Now you find due to price changes, for any goods sold on day i, if its direct benefit is val, the total benefit would be i⋅val.

Beginning from the first day, you can and must sell only one good per day until you can't or don't want to do so. If you are allowed to leave some goods unsold, what's the max total benefit you can get in the end?

 

Input

The first line contains an integers n(1≤n≤1000).
The second line contains n integers val1,val2,..,valn(−100≤val_i≤100).
The third line contains n integers cnt1,cnt2,..,cntn(1≤cnt_i≤100).

 

Output

Output an integer in a single line, indicating the max total benefit.

 

Sample Input

4
-1 -100 5 6
1 1 1 2

Sample Output

51

Hint

sell goods whose price with order as -1, 5, 6, 6, the total benefit would be -1*1 + 5*2 + 6*3 + 6*4 = 51.

 

Source

“浪潮杯”山东省第八届ACM大学生程序设计竞赛（感谢青岛科技大学）

 

思路:

我们可以先把所有商品的价值存在数组中（包括数量大于一的），排序并求出当前的 sum,ans

然后从数组最右边开始枚举，判断是否需要加上当前商品，若加上，则 sum += ans + arr[i]，因为总天数加了一天，所以后面的所有商品价值和也应该增加。

判断如果当前得到新的sum变小了，跳出，输出最大的 sum

#include<algorithm>
#include<iostream>
#include<cstdio>
using namespace std;
struct Goods{
    long long int val,cnt;
    bool friend operator < (Goods a,Goods b){
        return a.val < b.val;
    }
}goods[1005];
long long int n,flag,sum,ans,idx,arr[100005];
int main(){
    scanf("%lld",&n);
    for(int i = 1;i <= n;i++)
        scanf("%lld",&goods[i].val);
    for(int i = 1;i <= n;i++)
        scanf("%lld",&goods[i].cnt);
    sort(goods + 1,goods + n + 1);
    for(int i = 1;i <= n;i++){
        if(goods[i].cnt != 1)
            for(int j = 1;j <= goods[i].cnt;j++)
                arr[++idx] = goods[i].val;
        else
            arr[++idx] = goods[i].val;
    }
    flag = idx;
    for(int i = 1;i <= idx;i++)
        if(arr[i] >= 0){
            flag = i - 1;
            for(int k = i;k <= idx;k++){
                ans += arr[k];
                sum += arr[k] * (k - i + 1);
            }
            break;
        }
    if(flag == idx)
        printf("%lld\n",arr[idx]);
    else{
        for(int i = flag;i >= 1;i--){
            if(sum + ans + arr[i] > sum){
                sum += ans + arr[i];
                ans += arr[i];
            }
            else
                break;
        }
        printf("%lld\n",sum);
    }
    return 0;
}