2023年10月11日-秋招-第二题（200分）-塔子哥读书

在线评测链接

题目描述

  塔子哥是个爱学习的孩子，并且喜好读书，并且更喜欢读好书，在某一天他要读n本书，读每一本书都要花费1单位时间，并且塔子哥对每一本书有一个满意值，同时读完每一本书塔子哥都会收获一定的快乐值，现在塔子哥想知道要按照怎样的顺序读书读哪些书能使他收获的快乐值总和最大

  每本书籍的快乐值为读这本书和之前所有书所花费的时间乘以对这本书的满意值，可按照任意顺序读书，也可以放弃读某些书。

设书的总数为n，书的满意值为m

$1\le n\le 500$

$-1000\le m\le 1000$

输入描述

输入为字符串，字符串内容为代表每本书籍的满意值，例如输入为"4,3,2"，表示有3本书，满意值分别是4、3和2

输出描述

输出为数字，代表最大的快乐值之和

样例

输入1

4,3,2

输出1

20

说明

按照原来顺序相反的时间阅读书籍(2 * 1 +3 * 2+4 * 3=20)，满意值越高的书籍越后面读可以获得最大的快乐值。

输入2

-1,-4,-5

输出2

0

说明

这些书籍满意值都很低，所以不阅读任何书籍可以获得最大的快乐值 。

输入3

-1,-8,0,5,-9

输出3

14

说明

去掉第二个和最后一本书籍，最大的快乐时间系数和为(-1 * 1+0 * 2+5 * 3=14)每书籍都需要花费1单位时间来阅读 。

思路：贪心

其实本质上就是最大化$\sum i\times a[i]$

首先，如果$a[i]\le 0$，则可以直接舍弃$a[i]$

对于正整数的情况，一定是优先把数值低的放在前面，数值高的放在后面，即直接对原数组进行升序排列即可。

时间复杂度

$O(nlogn)$

代码

C++

#include<bits/stdc++.h>
using namespace std;
int main (){
    string line;
    getline(cin, line);
    stringstream ss(line);
    string token;
    vector<int> a;
    while (std::getline(ss, token, ',')) {
    	int x = stoi(token);
    	a.push_back(x);
    }
	sort(a.begin() , a.end());
	int t = 1;
	int ans = 0;
	int n = a.size();
	for (int i = 0 ; i < n ; i++){
		int t = 1 , now = 0;
		for (int j = i ; j < n ; j++){
			now += t * a[j];
			t++;
			ans = max(ans , now);
		}
	}
	cout << ans << endl;
	return 0;
}

python代码

books = list(map(int,input().split(",")))
books.sort()
num_book = len(books)
ret = 0
for i in range(num_book):
    ret += books[i]*(i+1)
for i in range(num_book):
    books.pop(0)
    new_ret = 0
    for j in range(num_book-i-1):
        new_ret += books[j]*(j+1)
    ret = max(ret,new_ret)
print(ret)

Java代码

import java.util.Arrays;
import java.util.Scanner;
import java.util.stream.IntStream;

public class Main {

	@SuppressWarnings({ "resource" })
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		String[] strs = sc.nextLine().split(",");
		int[] arr = Arrays.stream(strs).mapToInt(Integer::parseInt).toArray();
		Arrays.sort(arr);
		int[] sum = new int[arr.length];
		int res = 0;
		for (int i = 0; i < arr.length; i++) {
			sum[i] = IntStream.of(Arrays.copyOfRange(arr, i, arr.length)).sum();
			if (sum[i] > 0) {
				res = res + sum[i];
			}
		}
		System.out.println(res);
	}
}