You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
52 lines
1.7 KiB
52 lines
1.7 KiB
package class_2022_03_3_week;
|
|
|
|
import java.util.Arrays;
|
|
|
|
// 来自银联编程比赛
|
|
// 某公司计划推出一批投资项目。 product[i] = price 表示第 i 个理财项目的投资金额 price 。
|
|
// 客户在按需投资时，需要遵循以下规则：
|
|
// 客户在首次对项目 product[i] 投资时，需要投入金额 price
|
|
// 对已完成首次投资的项目 product[i] 可继续追加投入，
|
|
// 但追加投入的金额需小于上一次对该项目的投入(追加投入为大于 0 的整数)
|
|
// 为控制市场稳定，每人交易次数不得大于 limit。(首次投资和追加投入均记作 1 次交易)
|
|
// 若对所有理财项目中最多进行 limit 次交易，使得投入金额总和最大，请返回这个最大值的总和。
|
|
// 测试链接 : https://leetcode-cn.com/contest/cnunionpay-2022spring/problems/I4mOGz/
|
|
public class Code06_FinancialProduct {
|
|
|
|
public static long mod = 1000000007L;
|
|
|
|
public int maxInvestment(int[] arr, int limit) {
|
|
Arrays.sort(arr);
|
|
int n = arr.length;
|
|
long ans = 0;
|
|
int r = n - 1;
|
|
int l = r;
|
|
while (limit > 0 && r != -1) {
|
|
while (l >= 0 && arr[l] == arr[r]) {
|
|
l--;
|
|
}
|
|
int big = arr[r];
|
|
int small = l == -1 ? 0 : arr[l];
|
|
int teams = n - l - 1;
|
|
int all = (big - small) * teams;
|
|
if (limit >= all) {
|
|
ans += get(big, small + 1, teams);
|
|
ans %= mod;
|
|
limit -= all;
|
|
} else {
|
|
int a = limit / teams;
|
|
ans += get(big, big - a + 1, teams) + (long) (big - a) * (long) (limit % teams);
|
|
ans %= mod;
|
|
limit = 0;
|
|
}
|
|
r = l;
|
|
}
|
|
return (int) (ans % mod);
|
|
}
|
|
|
|
public static long get(long up, long down, long num) {
|
|
return num * ((up + down) * (up - down + 1) / 2);
|
|
}
|
|
|
|
}
|