E2. Rudolf and Snowflakes (hard version)

Codeforces Round 883 (Div. 3)E2

题意：给定n个顶点，试确定这n个点是否可以组成雪花。

雪花的定义：

1.雪花的第一层只有一个点，

2.第二层有k个点

3.第三层为第二层的每个点再连k个点（雪花至少为3层，第四层为第三层的每个点在连k个点，以此类推

数据范围是n<=1e18,k>1

根据上述条件，我们可以发现如果可以构成一个雪花，那么该雪花中点的个数是1+k+k^2+k^3+...+k^p,根据数据范围，如果我们写两重循环，第一重是k，k的范围是2~1e6，第二重是p，因为2^62大于1e18，所以p的范围是p<=62

上述的k的范围，

1.如果我们直接存入最小的值是1+k+k^2，那么k的范围是k<1e9,此时时间复杂度是O(1e9*62)

2.如果我们直接存入最小的值是1+k+k^2+k^3，那么k的范围是k<1e6,时间复杂度是O(1e6*62)

那么就需要特判一下是否1+k+k^2=n,是否可以解出一个正整数k，且k大于1.

#include<bits/stdc++.h>
using namespace std;
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
//#define int long long
//#define x first
//#define y second

typedef long long LL;
const int N = 2e5+10;
unordered_set<long long> se;
 
void solve()
{
	LL n;
	cin>>n;
	
	LL init=4*n-3;
	double m=sqrt(init);
	LL y;
	y=(LL)m;
	if(y*y==init&&(y-1)%2==0&&(y-1)/2>1)
	{
		cout<<"YES"<<endl;
		return;
	}
	if(!se.count(n)) cout<<"NO"<<endl;
	else cout<<"YES"<<endl;
}

int main()
{
	IOS; 
	for(LL k=2;k<=1000000;k++)
	{
		LL val=1+k+k*k;
		LL p=k*k*k;
		for(LL cnt=3;cnt<63;cnt++)
		{
			val+=p;
			if(val>1e18) break;
			se.insert(val);
			if(p>(1e18)/k) break;
			p*=k;
		}
	}
	int t;
	cin>>t;
	while(t--) 
    solve();
	return 0;
}