結果

問題 No.2979 直角三角形の個数
ユーザー kotatsugame
提出日時 2024-12-25 06:42:56
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,958 ms / 4,000 ms
コード長 1,101 bytes
コンパイル時間 657 ms
コンパイル使用メモリ 66,340 KB
実行使用メモリ 7,680 KB
最終ジャッジ日時 2024-12-25 06:43:12
合計ジャッジ時間 7,640 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<cassert>
using namespace std;
long N;
int pf[1<<20];
int ps[7],T[1<<7];
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	for(int i=2;i<1<<20;i++)if(pf[i]==0)
	{
		for(int j=i;j<1<<20;j+=i)pf[j]=i;
	}
	cin>>N;
	long ans=0;
	for(int m=2;;m++)
	{
		long up=(double)N/(2*m);
		int l=m+1,r=(int)min(up,2*m-1L);
		if(l>r)break;
		int pssz=0;
		{
			int v=m;
			while(v%2==0)v/=2;
			while(v>1)
			{
				int p=pf[v];
				ps[pssz++]=p;
				while(v%p==0)v/=p;
			}
		}
		assert(pssz<=6);
		long sum=0;
		T[0]=1;
		for(int i=0;i<1<<pssz;i++)
		{
			if(i)
			{
				int j=0;
				while(!(i>>j&1))j++;
				T[i]=T[i^1<<j]*ps[j];
			}
			int t=T[i];
			int nl=(double)(l+t-1)/t,nr=(double)r/t;
			if(nl%2==0)nl++;
			long nup=(double)up/t;
			long cur=0;
			while(nl<=nr)
			{
				long q=(double)nup/nl;
				int nnl=nr;
				if(nup<q*nr)nnl=(double)nup/q;
				if(nnl%2==0)nnl--;
				//int nnl=(int)min((long)nr,(long)((double)nup/q))+1;
				cur+=q*((nnl-nl)/2+1);
				nl=nnl+2;
			}
			if(__builtin_parity(i))sum-=cur;
			else sum+=cur;
		}
		ans+=sum;
	}
	cout<<ans<<endl;
}
0