21.开灯问题

#include<stdio.h>

#include<string.h>

int light(int m,int k[],int n)

{
int i = 1;
if(m == 1)
{
for(i=1;i<=n;i++)
{
k[i] = 1;
}
}
else if(m == 2)
{
for(i=1;i<=n;i++)
{
if(i%m == 0)
{
k[i] = 0;
}
}
}
else
{
for(i=1;i<=n;i++)
{
if(i%m == 0)
{
if(k[i] == 0)
{
k[i] = 1;
}
else
{
k[i] = 0;
}
}
}
}
return 0;

}

int main()

{
int n,k,i = 1;
int a[1001]={0};
memset(a,0,1001);
scanf("%d %d",&n,&k);
while(i <= k)
{
light(i,a,n);
i++;
}
i = 1;
while(i <= n)
{
if(a[i])
{
printf("%d ",i);
}
i++;
}
printf("\n");
return 0;

}