A13085 | K for the Price of One (Hard Version)
时间限制1s
内存限制256MB
通过 / 提交0/0
题目描述
This is the hard version of this problem. The only difference is the constraint on $k$ — the number of gifts in the offer. In this version: $2 \le k \le n$ .
Vasya came to the store to buy goods for his friends for the New Year. It turned out that he was very lucky — today the offer " $k$ of goods for the price of one" is held in store.
Using this offer, Vasya can buy exactly $k$ of any goods, paying only for the most expensive of them. Vasya decided to take this opportunity and buy as many goods as possible for his friends with the money he has.
More formally, for each good, its price is determined by $a_i$ — the number of coins it costs. Initially, Vasya has $p$ coins. He wants to buy the maximum number of goods. Vasya can perform one of the following operations as many times as necessary:
- Vasya can buy one good with the index $i$ if he currently has enough coins (i.e $p \ge a_i$ ). After buying this good, the number of Vasya's coins will decrease by $a_i$ , (i.e it becomes $p := p - a_i$ ).
- Vasya can buy a good with the index $i$ , and also choose exactly $k-1$ goods, the price of which does not exceed $a_i$ , if he currently has enough coins (i.e $p \ge a_i$ ). Thus, he buys all these $k$ goods, and his number of coins decreases by $a_i$ (i.e it becomes $p := p - a_i$ ).
Please note that each good can be bought no more than once.
For example, if the store now has $n=5$ goods worth $a_1=2, a_2=4, a_3=3, a_4=5, a_5=7$ , respectively, $k=2$ , and Vasya has $6$ coins, then he can buy $3$ goods. A good with the index $1$ will be bought by Vasya without using the offer and he will pay $2$ coins. Goods with the indices $2$ and $3$ Vasya will buy using the offer and he will pay $4$ coins. It can be proved that Vasya can not buy more goods with six coins.
Help Vasya to find out the maximum number of goods he can buy.
Vasya came to the store to buy goods for his friends for the New Year. It turned out that he was very lucky — today the offer " $k$ of goods for the price of one" is held in store.
Using this offer, Vasya can buy exactly $k$ of any goods, paying only for the most expensive of them. Vasya decided to take this opportunity and buy as many goods as possible for his friends with the money he has.
More formally, for each good, its price is determined by $a_i$ — the number of coins it costs. Initially, Vasya has $p$ coins. He wants to buy the maximum number of goods. Vasya can perform one of the following operations as many times as necessary:
- Vasya can buy one good with the index $i$ if he currently has enough coins (i.e $p \ge a_i$ ). After buying this good, the number of Vasya's coins will decrease by $a_i$ , (i.e it becomes $p := p - a_i$ ).
- Vasya can buy a good with the index $i$ , and also choose exactly $k-1$ goods, the price of which does not exceed $a_i$ , if he currently has enough coins (i.e $p \ge a_i$ ). Thus, he buys all these $k$ goods, and his number of coins decreases by $a_i$ (i.e it becomes $p := p - a_i$ ).
Please note that each good can be bought no more than once.
For example, if the store now has $n=5$ goods worth $a_1=2, a_2=4, a_3=3, a_4=5, a_5=7$ , respectively, $k=2$ , and Vasya has $6$ coins, then he can buy $3$ goods. A good with the index $1$ will be bought by Vasya without using the offer and he will pay $2$ coins. Goods with the indices $2$ and $3$ Vasya will buy using the offer and he will pay $4$ coins. It can be proved that Vasya can not buy more goods with six coins.
Help Vasya to find out the maximum number of goods he can buy.
输入格式
The first line contains one integer $t$ ( $1 \le t \le 10^4$ ) — the number of test cases in the test.
The next lines contain a description of $t$ test cases.
The first line of each test case contains three integers $n, p, k$ ( $2 \le n \le 2 \cdot 10^5$ , $1 \le p \le 2\cdot10^9$ , $2 \le k \le n$ ) — the number of goods in the store, the number of coins Vasya has and the number of goods that can be bought by the price of the most expensive of them.
The second line of each test case contains $n$ integers $a_i$ ( $1 \le a_i \le 10^4$ ) — the prices of goods.
It is guaranteed that the sum of $n$ for all test cases does not exceed $2 \cdot 10^5$ .
The next lines contain a description of $t$ test cases.
The first line of each test case contains three integers $n, p, k$ ( $2 \le n \le 2 \cdot 10^5$ , $1 \le p \le 2\cdot10^9$ , $2 \le k \le n$ ) — the number of goods in the store, the number of coins Vasya has and the number of goods that can be bought by the price of the most expensive of them.
The second line of each test case contains $n$ integers $a_i$ ( $1 \le a_i \le 10^4$ ) — the prices of goods.
It is guaranteed that the sum of $n$ for all test cases does not exceed $2 \cdot 10^5$ .
输出格式
For each test case in a separate line print one integer $m$ — the maximum number of goods that Vasya can buy.
输入输出样例
输入 #1
8 5 6 2 2 4 3 5 7 5 11 2 2 4 3 5 7 3 2 3 4 2 6 5 2 3 10 1 3 9 2 2 10000 2 10000 10000 2 9999 2 10000 10000 4 6 4 3 2 3 2 5 5 3 1 2 2 1 2
输出 #1
3 4 1 1 2 0 4 5
暂无题解
C++ 编辑器
输入
输出
可保存默认模板;新题优先使用已保存模板。
当前快捷键仅展示,暂不支持修改。
- 撤销
Ctrl / ⌘ + Z - 重做
Ctrl / ⌘ + Y - 查找
Ctrl / ⌘ + F - 全选
Ctrl / ⌘ + A - 复制
Ctrl / ⌘ + C - 剪切
Ctrl / ⌘ + X - 粘贴
Ctrl / ⌘ + V - 自动排版
工具栏排版按钮 - 草稿保存
编辑时自动保存到本机
历史
提交记录
状态说明时间源码
AI
作答助手
你好,我是作答助手。可以问思路、复杂度、样例含义或代码报错原因;不会直接给出完整 AC 代码。
确定要清空代码吗?
提交通过
评测结果:Accepted