10pts求助
查看原帖
10pts求助
1002517
Adorable_楼主2024/7/18 16:03

五颜六色的

#include<bits/stdc++.h>
using namespace std;

#define int long long

bool yes(int x)
{
	if(sqrt(x) * sqrt(x) == x) return 1;
	return 0;
}

const int N = 7e4+5;
int ans = 0,k,n,a[N];

signed main()
{
	cin>>n>>k;
	for(int i = 1;i<=n;++i) a[i] = i+a[i-1];
	for(int i = 1;i<=n;++i)
	{
		int j = i+k-1;
		if(yes(a[j]-a[i-1]))
			ans++;
	}
	cout<<ans<<endl;
	return 0;
}
2024/7/18 16:03
加载中...