Educational Codeforces Round 127 (Rated for Div. 2) C. Dolce Vita

题目

C. Dolce Vita

思路

观察样例一，可以发现 :

下标	1	2	3
第1天	a	b	c
第2天	a	b
第3天	a	b
第4天	a
第5天	a
第6天	a
第7天	a

可以归纳出: 想要统计总共买了多少个物品，只需分别统计 ：买到前1个物品的天数, 买到前2个物品的天数, …, 买到前i个物品的天数.
对于上面的表格：
买到前1个的天数是8, 买到前2个的天数是3, 买到前3个的天数是1. 所以总共是$7+3+1=11$个.

假设用$k$表示增加的价格(从第二天开始每天增加1), $x$表示每天拥有的钱

考虑买前1个，那么有 $a_1+k\le x$, 则$k\le x-a_1$, 这意味着可以从第$0$天到第$k=x-a_1$天可以买物品$a_1$, 所以天数$d=k+1=x-a_1+1$.

考虑买前2个， 那么有 $a_1+a_2+2k\le x$, 则$2k\le x-(a_1+a_2)$, 这意味着可以从第$0$天到第$k=\frac{x-(a_1+a_2)}{2}$天可以买物品$a_1$, 所以天数$d=k+1=\frac{x-(a_1+a_2)}{2}+1$.

考虑买前i个，那么有 (${\sum }_{j=1}^i{a}_j)+ik\le x$, 则$ik\le x-({\sum }_{j=1}^i{a}_j)$, 这意味着可以从第$0$天到第$k=\frac{x-{\sum }_{j=1}^i{a}_j}{i}$天可以买物品$a_1$, 所以天数: $d=k+1=\frac{x-{\sum }_{j=1}^i{a}_j}{i}+1$.

注意：在计算开始前要将数组排序，优先考虑便宜的

实现

#include <iostream>
#include <algorithm>
 
using namespace std;
 
const int N = 2e5 + 10;
int n, x;
int a[N];
 
int main()
{
    int T;
    cin >> T;
    while(T--)
    {
        cin >> n >> x;
        for(int i = 1; i <= n; i ++) cin >> a[i];
 
        sort(a + 1, a + 1 + n);
 
        long long sum = 0, tol = 0;
        for(int i = 1; i <= n; i ++)
        {
            tol += a[i];
            if(x - tol >= 0) sum += (x - tol) / i + 1;
        }
 
        cout << sum << endl;
    }
    
    return 0;
}