HDU—1205—吃糖果—【入门题】

吃糖果

Time Limit: 6000/3000 MS (Java/Others)    Memory Limit: 65535/32768 K (Java/Others)
Total Submission(s): 23202    Accepted Submission(s): 6663

Problem Description

HOHO，终于从Speakless手上赢走了所有的糖果，是Gardon吃糖果时有个特殊的癖好，就是不喜欢将一样的糖果放在一起吃，喜欢先吃一种，下一次吃另一种，这样；可是Gardon不知道是否存在一种吃糖果的顺序使得他能把所有糖果都吃完？请你写个程序帮忙计算一下。

 

Input

第一行有一个整数T，接下来T组数据，每组数据占2行，第一行是一个整数N（0<N<=1000000)，第二行是N个数，表示N种糖果的数目Mi(0<Mi<=1000000)。

 

Output

对于每组数据，输出一行，包含一个"Yes"或者"No"。

 

Sample Input

  
  

   
   2
3
4 1 1
5
5 4 3 2 1
  
  

 

Sample Output

  
  

   
   No
Yes


   
   

    
    

     
     Hint
    
    
Hint
   
   
   
    
Please use function scanf
  
  

 

我的想法很简单【虽然不能证明它的正确性】:

比如有5个同一类的糖果：A A A A A ，要求不能连续吃到两个A，那就在5个A的4个空隙里放入4个不同类的糖果。

以此类推，第i类糖果有sweet[i]个，共有sweet[i]-1个空隙，总共sum个糖果，那么sum-sweet[i] 就是其他类的糖果数目，要把空隙填满，那么就不能少于sweet[i]-1个其他类的糖果，所以判断的关键就是 sum-sweet[i] >= sweet[i]-1

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
using namespace std;
#define LL __int64
int sweet[1000010];

int main()
{
    int cases;
    scanf("%d",&cases);
    while(cases--)
    {
        int m;
        LL sum=0;       //sum要__int64 ，因为m<=10^6 ， Mi<=10^6 ， 最坏情况就是sum<=10^12 ， 最大值会超过int【略大于2.1*10^9】
        scanf("%d",&m);
        for(int i=0;i<m;i++)
        {
            scanf("%d",sweet+i);
            sum+=sweet[i];
        }
        int sign=1;
        for(int i=0;i<m;i++)
            if(sum-sweet[i]<sweet[i]-1)
            {
                sign=0;
                break;
            }
        if(sign)
            printf("Yes\n");
        else
            printf("No\n");
    }
    return 0;
}

优化下代码，思路基本不变，因为总会有一类或几类的糖果数目最多，那么就一定能填满其他类糖果的空隙。。。所以只需要判断数目最多的糖果的空隙能不能被填满。。。

这样也不用开数组了。。。只要用maxn变量存放最多的某类糖果数目。。。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
using namespace std;
#define LL __int64
int main()
{
    int cases;
    scanf("%d",&cases);
    while(cases--)
    {
        int m,temp,maxn=0;
        LL sum=0;       //sum要__int64 ，因为m<=10^6 ， Mi<=10^6 ， 最坏情况就是sum<=10^12 ， 最大值会超过int【略大于2.1*10^9】
        scanf("%d",&m);
        for(int i=0;i<m;i++)
        {
            scanf("%d",&temp);
            sum+=temp;
            if(temp>maxn)
                maxn=temp;
        }
        if(sum-maxn>=maxn-1)
            printf("Yes\n");
        else
            printf("No\n");
    }
    return 0;
}