2018 计蒜之道 初赛 第六场 贝壳找房

#include <bits/stdc++.h>
using namespace std;
const int N=1e4+5; 
int n,m;
int a[N];
int main()
{
memset(a,0,sizeof(a)); 
scanf("%d",&n);
while(n--)
{

scanf("%d",&m);
for(int i=0;i<m;i++)
{
int tp; scanf("%d",&tp); 
   a[i]=tp;
}
sort(a,a+m); int flag=1,jud=1,j=0; 
            int k=1;
while(j<m)
{
if(jud<a[j])
{
jud=jud+k+2; k++;
}
if(jud>a[j])
{
flag=0; break;
}
if(jud==a[j]&&a[j+1]!=a[j])
{
j++; jud=jud+k+2; k++;
}else if(jud==a[j])
       j++;
}  
if(flag)printf("yes\n");
else printf("no\n");
}
return 0;
 }