bzoj5387 [Lydsy1806月赛]质数拆分 乱搞

Description

---

给定一个正整数n求a+b+c+d=n的解的个数，其中a、b、c、d都为质数且可以相同
n<=150000

Solution

---

一眼题
容易想到先求a+b的所有解扔进map里面然后枚举c+d，这样就能获得TLE的好成绩
显然我们统计了两个质数之和的方案数后答案$=\sum_{i=1}^{n}map[i]*map[n-i]$，这样就能获得TLE的好成绩
然后随意卡卡常数什么的就过了

Code

---

#include <stdio.h>

typedef long long LL;
const int N=150005;

int prime[N/10],map[N+5],cnt;
bool not_prime[N];
int n,x,i,j;
LL ans=0;

__attribute__((optimize("O2")))
int main(void) {
    scanf("%d",&n);
    for (i=2;i<=n;i++) {
        if (!not_prime[i]) prime[++cnt]=i;
        for (j=1;j<=cnt&&i*prime[j]<=n;j++) {
            not_prime[i*prime[j]]=1;
            if (i%prime[j]==0) break;
        }
    }
    for (i=1;i<=cnt&&(prime[i]<<1)<=n;i++) {
        for (j=i+1;j<=cnt&&prime[i]+prime[j]<=n;j++) {
            map[prime[i]+prime[j]]+=2;
        }
        map[prime[i]<<1]++;
    }
    for (i=(n>>1);i;i--) ans+=map[i]*map[n-i]*2;
    if (!(n&1)) ans-=map[n>>1]*map[n>>1];
    printf("%lld", ans);
    return 0;
}