XDOJ 期末练习题 数字统计

#include<stdio.h>
int main()
{
int n=0,k=0,num[201]={1,2,3},zheng=0,yu1=0,yuda=0;
scanf("%d %d",&n,&k) ;
int i=3;
while(i<=k)
{
num[i]=(i-1)*(i-1)+3*(i-2)+1;
i++;
}
i=1;
while(i<=k)
{
if(num[i]%n==0){zheng++;
}
if(num[i]%n==1){yu1++;
}
if(num[i]%n>1){yuda++;
}
i++;
}
printf("%d %d %d",zheng,yu1,yuda);


	return 0; 
}