两种开灯问题

题目描述
有n盏灯，编号为1～n。第1个人把所有灯打开，第2个人按下所有编号为2 的倍数的开关（这些灯将被关掉），第3个人按下所有编号为3的倍数的开关（其中关掉的灯 将被打开，开着的灯将被关闭），依此类推。一共有k个人，问最后开着的灯的数量是多少？

输入
输 入两个正整数n和k（k≤n≤100000），用空格隔开。

输出
输出开着的灯的数量。

**样例输入
5 3
样例输出
2

直接两层循环

#include<bits/stdc++.h>
using namespace std;
int main()
{
	int n,k;
	cin>>n>>k;
	bool arr[n+5];
	memset(arr,true,sizeof arr);
	for(int i=2;i<=k;i++){
		for(int j=i;j<=n;j+=i){
			arr[j]=!arr[j];
		}
	}
	int res=0;
	for(int i=1;i<=n;i++){
		if(arr[i])res++;
	}
	cout<<res;
	return 0;
}

着重说说第二种开灯问题的O(n)解法
第二种问题就是，n个灯，n个人每个人操作一遍，也就是上述问题中k=n的情况
此时稍加思考可以发现，题目可以转换为求每一个位置下标的因子个数问题
一个数的因子小于等于自身，而且形成一种对称关系，也就是说如果数字 i 是完全平方数，即 t*t=i ，那么它的因子个数就是奇数个，也就是说从第一个开始，它被操作奇数次，同理非完全平方数被操作偶数次。

#include<bits/stdc++.h>
using namespace std;
int main()
{
	int n,k;
	cin>>n>>k;
	int res=0;
	for(int i=1;i<=k;i++){
		int t=sqrt(i);
		if(t*t==i){
			res++;
		}
	}
	cout<<res;
	return 0;
}