Sport Mafia
Hazrat Ali
Alya puts candies into the box, which will serve as a prize for a winner. To do that, she performs n actions. The first action performed is to put a single candy into the box. For each of the remaining moves she can choose from two options:
- the first option, in case the box contains at least one candy, is to take exactly one candy out and eat it. This way the number of candies in the box decreased by 1;
- the second option is to put candies in the box. In this case, Alya will put 1 more candy, than she put in the previous time.
Thus, if the box is empty, then it can only use the second option.
For example, one possible sequence of Alya's actions look as follows:
- put one candy into the box;
- put two candies into the box;
- eat one candy from the box;
- eat one candy from the box;
- put three candies into the box;
- eat one candy from the box;
- put four candies into the box;
- eat one candy from the box;
- put five candies into the box;
This way she will perform 9 actions, the number of candies at the end will be 11, while Alya will eat 4 candies in total.
You know the total number of actions n and the number of candies at the end k. You need to find the total number of sweets Alya ate. That is the number of moves of the first option. It's guaranteed, that for the given n and k the answer always exists.
Please note, that during an action of the first option, Alya takes out and eats exactly one candy.
The first line contains two integers n and k (1≤n≤10; 0≤k≤10) — the total number of moves and the number of candies in the box at the end.
It's guaranteed, that for the given n and k the answer exists.
Print a single integer — the number of candies, which Alya ate. Please note, that in this problem there aren't multiple possible answers — the answer is unique for any input data.
1 1
0
9 11
4
5 0
3
3 2
1
Solution
#include <bits/stdc++.h>
using namespace std;
int main()
{
long long n, k;
cin >> n >> k;
long long l = 0, h = n;
while (l < h)
{
long long m = (l + h) / 2;
long long temp = m * (m + 1) / 2;
if (temp + m < k + n)
{
l = m + 1;
}
else
{
h = m;
}
}
cout << n - l << endl;
return 0;
}