B. Brightness Begins Codeforces Round 976 (Div. 2)

原题

B. Brightness Begins

 

解析

Hint 1

第 i 个灯泡最终状态与 n 的大小无关

Hint 2

第 i 个灯泡最终状态与 i 的约数数量的奇偶性相关

Solution

对任意灯泡 i , 它的最终状态由其约数数量的奇偶性相关, 如果 i 有偶数个约数, 那么会是亮的, 否则会是暗的. 换句话说, 如果 i 不是一个完全平方数, 这个灯泡就会是亮的, 否则就是暗的.

因此问题转换成了找到第 k 个不是完全平方数的数这样一个数学问题.

可以用二分搜索 n -  = k

也可以直接算 k +  + 0.5

具体原因, 作者不是很清楚, 可能是某种数学推导证出来的, 因此作者打算熟记第k个不是完全平方数的数是k +  + 0.5即可

sqrtl

当处理 long long 类型的求根时用 sqrtl 会更准确

代码

#include <bits/stdc++.h>
#define int long long

using namespace std;

const int N = 200010;

int n, m, k, q, ans;

int a[N];

void solve()
{
	cin >> k;
	cout << k + (long long) (sqrtl(k) + 0.5) << "\n";
}

signed main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	int T = 1;
	cin >> T;
	while (T -- )
	{
		solve();
	}
}