1603: 大小鱼（PIPIPOJ）

1603: 大小鱼（PIPIPOJ）

题目描述

PIPI现在有a条大鱼和b条小鱼，以及一个容量为c的鱼缸，当放入一条大鱼后，容器内的小鱼都会被吃掉。
但是当放入大鱼后再放入小鱼，则小鱼不会被吃掉，需要重新加入大鱼才能吃掉这些新放入的小鱼
PIPI想把这些鱼都放到鱼缸里，请问他是否能做到。
大鱼和小鱼的体积都是1，且大鱼不会被吃掉。

输入

多组输入。
每组输入题中描述的三个整数a,b,c(1<=a,c<=1e9,1<=b<=1e18)。

输出

对于每组输入，若能将所有鱼都放入容器中，输出YES，否则输出NO。

样例输入

6 10 7
2 5 3

样例输出

YES
NO

题解1（C++版本）

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
  
LL a, b, c;
  
int main(){
    while(~scanf("%lld%lld%lld", &a,&b,&c)){
        bool flag = true;
        if(a <= c){ // 所有大鱼的体积之和小于或等于鱼缸的体积
            b -= (a * c - (1LL + a) * a / 2); // 先放c-1条小鱼，再放1条大鱼，等大鱼吃掉之前放的小鱼后，鱼缸内只剩1条大鱼，此时鱼缸体积为c-1；接着放c-2条小鱼，再放1条大鱼，等大鱼吃掉之前放的小鱼后，鱼缸内只剩2条大鱼，此时鱼缸体积为c-2；由于大鱼的数量为a条，重复这个步骤a次后，放入鱼缸的大鱼共计a条，放入的小鱼共计(c-1)+(c-2)+...+1 = a(1+a)/2，鱼缸空闲的体积为c-a，还有c-a(1+a)/2条小鱼没有放入鱼缸，因此要判断c-a(1+a)/2 是否大于 c-a
            c -= a;
            if(b > c) flag = false;
        }else flag = false;
        if(flag == true) printf("YES\n");
        else printf("NO\n");
          
    }   
    return 0;
}