hdu 2060Snooker

Wrong Code

#include"stdio.h"
int main()
{
int t,n,a,b,sum;
scanf("%d",&t);
while(t--)
{
scanf("%d%d%d",&n,&a,&b);
sum=a;
if(n>=6)
sum+=(n-6)*1+2+3+4+5+6+7;
else
sum+=(n+1+2)*n/2;
if(sum>=b)
printf("Yes\n");
else
printf("No\n");
}
return 0;
}

AC Code

#include<stdio.h>
#include<string.h>
#include<math.h>
int fen[]={0,7,13,18,22,25,27};
int fun(int x)
{
if(x<=6) return fen[x];
else return (8*(x-6)+fen[6]);
}
int main()
{
int T;
int a,b,c;
scanf("%d",&T);
while(T--)
{
scanf("%d%d%d",&a,&b,&c);
if(b+fun(a)>=c) printf("Yes\n");
else printf("No\n");
}
return 0;
}