A16093 | Kolya and Movie Theatre
时间限制1s
内存限制256MB
通过 / 提交0/0
题目描述
Recently, Kolya found out that a new movie theatre is going to be opened in his city soon, which will show a new movie every day for $n$ days. So, on the day with the number $1 \le i \le n$ , the movie theatre will show the premiere of the $i$ -th movie. Also, Kolya found out the schedule of the movies and assigned the entertainment value to each movie, denoted by $a_i$ .
However, the longer Kolya stays without visiting a movie theatre, the larger the decrease in entertainment value of the next movie. That decrease is equivalent to $d \cdot cnt$ , where $d$ is a predetermined value and $cnt$ is the number of days since the last visit to the movie theatre. It is also known that Kolya managed to visit another movie theatre a day before the new one opened — the day with the number $0$ . So if we visit the movie theatre the first time on the day with the number $i$ , then $cnt$ — the number of days since the last visit to the movie theatre will be equal to $i$ .
For example, if $d = 2$ and $a = [3, 2, 5, 4, 6]$ , then by visiting movies with indices $1$ and $3$ , $cnt$ value for the day $1$ will be equal to $1 - 0 = 1$ and $cnt$ value for the day $3$ will be $3 - 1 = 2$ , so the total entertainment value of the movies will be $a_1 - d \cdot 1 + a_3 - d \cdot 2 = 3 - 2 \cdot 1 + 5 - 2 \cdot 2 = 2$ .
Unfortunately, Kolya only has time to visit at most $m$ movies. Help him create a plan to visit the cinema in such a way that the total entertainment value of all the movies he visits is maximized.
However, the longer Kolya stays without visiting a movie theatre, the larger the decrease in entertainment value of the next movie. That decrease is equivalent to $d \cdot cnt$ , where $d$ is a predetermined value and $cnt$ is the number of days since the last visit to the movie theatre. It is also known that Kolya managed to visit another movie theatre a day before the new one opened — the day with the number $0$ . So if we visit the movie theatre the first time on the day with the number $i$ , then $cnt$ — the number of days since the last visit to the movie theatre will be equal to $i$ .
For example, if $d = 2$ and $a = [3, 2, 5, 4, 6]$ , then by visiting movies with indices $1$ and $3$ , $cnt$ value for the day $1$ will be equal to $1 - 0 = 1$ and $cnt$ value for the day $3$ will be $3 - 1 = 2$ , so the total entertainment value of the movies will be $a_1 - d \cdot 1 + a_3 - d \cdot 2 = 3 - 2 \cdot 1 + 5 - 2 \cdot 2 = 2$ .
Unfortunately, Kolya only has time to visit at most $m$ movies. Help him create a plan to visit the cinema in such a way that the total entertainment value of all the movies he visits is maximized.
输入格式
Each test consists of multiple test cases. The first line contains a single integer $t$ ( $1 \le t \le 10^4$ ) — the number of test cases. The description of the test cases follows.
The first line of each test case contains three integers $n$ , $m$ , and $d$ ( $1 \le n \le 2 \cdot 10^5$ , $1 \le m \le n$ , $1 \le d \le 10^9$ ).
The second line of each set of input data contains $n$ integers $a_1, a_2, \ldots, a_n$ ( $-10^9 \le a_i \le 10^9$ ) — the entertainment values of the movies.
It is guaranteed that the sum of $n$ over all test cases does not exceed $2 \cdot 10^5$ .
The first line of each test case contains three integers $n$ , $m$ , and $d$ ( $1 \le n \le 2 \cdot 10^5$ , $1 \le m \le n$ , $1 \le d \le 10^9$ ).
The second line of each set of input data contains $n$ integers $a_1, a_2, \ldots, a_n$ ( $-10^9 \le a_i \le 10^9$ ) — the entertainment values of the movies.
It is guaranteed that the sum of $n$ over all test cases does not exceed $2 \cdot 10^5$ .
输出格式
For each test case, output a single integer — the maximum total entertainment value that Kolya can get.
输入输出样例
输入 #1
6 5 2 2 3 2 5 4 6 4 3 2 1 1 1 1 6 6 6 -82 45 1 -77 39 11 5 2 2 3 2 5 4 8 2 1 1 -1 2 6 3 2 -8 8 -2 -1 9 0
输出 #1
2 0 60 3 0 7
The first test case is explained in the problem statement.
In the second test case, it is optimal not to visit any movies.
In the third test case, it is optimal to visit movies with numbers $2$ , $3$ , $5$ , $6$ , so the total entertainment value of the visited movies will be $45 - 6 \cdot 2 + 1 - 6 \cdot 1 + 39 - 6 \cdot 2 + 11 - 6 \cdot 1 = 60$ .
In the second test case, it is optimal not to visit any movies.
In the third test case, it is optimal to visit movies with numbers $2$ , $3$ , $5$ , $6$ , so the total entertainment value of the visited movies will be $45 - 6 \cdot 2 + 1 - 6 \cdot 1 + 39 - 6 \cdot 2 + 11 - 6 \cdot 1 = 60$ .
C++ 编辑器
输入
输出
可保存默认模板;新题优先使用已保存模板。
当前快捷键仅展示,暂不支持修改。
- 撤销
Ctrl / ⌘ + Z - 重做
Ctrl / ⌘ + Y - 查找
Ctrl / ⌘ + F - 全选
Ctrl / ⌘ + A - 复制
Ctrl / ⌘ + C - 剪切
Ctrl / ⌘ + X - 粘贴
Ctrl / ⌘ + V - 自动排版
工具栏排版按钮 - 草稿保存
编辑时自动保存到本机
历史
提交记录
状态说明时间源码
AI
作答助手
你好,我是作答助手。可以问思路、复杂度、样例含义或代码报错原因;不会直接给出完整 AC 代码。
确定要清空代码吗?
提交通过
评测结果:Accepted