Double Happiness (筛素数 费马二平方定理 bitset优化)

双重幸福 - 代码强制 113C - 虚拟裁判 (csgrandeur.cn)

费马二平方定理:

所有素数可以表示为4k + 1,4k + 3(2除外)，表示为4k + 1的数可以被表示为a^2 + b ^2。

---

bitset优化:bitset - OI Wiki (oi-wiki.org)

题意: 区级[l,r]内有多少素数满足拆成两数平方和,记录总数

需要特判2

code:

#pragma GCC optimize(1)
#pragma GCC optimize(2)
#pragma GCC optimize(3,"Ofast","inline")
#include<bits/stdc++.h>
#define x first
#define y second
#define mak make_pair
#define IOS ios::sync_with_stdio(false),cin.tie(0),cout.tie(0)
#define debug(a) cout<<a<<'\n'
#define endl '\n'
#define umap unordered_map
using namespace std;
typedef long long LL;
typedef pair<int,int> PII;
const int N=3e8+5,M=1,inf=0x3f3f3f3f,mod=1e9+7;
int l,r;
int prime[20000005],cnt;
vector<bool> st(N);

void get_prime(int n)
{
    for(int i=2;i<=n;i++)
    {
        if(st[i]==false) prime[cnt++]=i;
        for(int j=0;prime[j]<=n/i;j++)
        {
            st[prime[j]*i]=true;
            if(i%prime[j]==0) break;
        }
    }
}

int main()
{
    cin>>l>>r;
    get_prime(r);
    int res=0;
    
    for(int i=0;i<cnt;i++)
    {
       if(prime[i]<l) continue;
       if(prime[i]%4==1) res++;
	}
    
    if(l<=2&&r>=2) res++;
    
    cout<<res<<endl;
    
	return 0;
}

END

---