Advertisement
Ankit_132

C

Jun 20th, 2024
520
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.26 KB | None | 0 0
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3. #define int long long
  4. signed main(int argc, char const *argv[])
  5. {
  6.     ios::sync_with_stdio(0);
  7.     cin.tie(0);
  8.     cout.tie(0);
  9.     int t;
  10.     cin >> t;
  11.     while (t--)
  12.     {
  13.         int n, c;
  14.         cin >> n >> c;
  15.         int total = 0;
  16.         vector<int> a(n);
  17.         for (auto &elem : a)
  18.         {
  19.             cin >> elem;
  20.             total += elem;
  21.         }
  22.         vector<int> v;
  23.         v.push_back(0);
  24.         for (int i = 1; i < n; i++)
  25.             v.push_back(a[i]);
  26.         sort(v.begin(), v.end());
  27.         v.push_back(0);
  28.         vector<int> pre(n + 1, 0);
  29.         vector<int> suf(n + 1, 0);
  30.  
  31.         for (int i = 1; i <= n - 1; i++)
  32.         {
  33.             pre[i] += pre[i - 1] + v[i];
  34.         }
  35.         for (int i = n - 1; i >= 1; i--)
  36.         {
  37.             suf[i] += suf[i + 1] + v[i];
  38.         }
  39.         int sum = a[0];
  40.         int ans = 1;
  41.         for (int i = 0; i <= n - 1; i++)
  42.         {
  43.             int cost = min((sum + pre[i]) * (total - sum - pre[i]), (sum + suf[n - i]) * (total - sum - suf[n - i]));
  44.             if (cost <= c)
  45.             {
  46.                 ans = i + 1;
  47.                 break;
  48.             }
  49.         }
  50.         cout << ans << "\n";
  51.     }
  52.     return 0;
  53. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement