NEFU 1321 差点是素数 (二分优化)

NEFU 1321

差点是素数
Problem:1321

Time Limit:1000ms

Memory Limit:65535K

Description
统计两个区间[l1,r1],[l2,r2]的整数中有多少个数满足：它本身不是素数，但只有一个素因子。当x两个区间都符合时，答案只算一次
Input
一共t组数据 t<=25005
每组数据l1,r1,l2,r2 (1<=l1<=r1<=1e12,1<=l2<=r2<=1e12)
Output
输出满足两个区间整数个数
Sample Input
1
1 5 2 10
Sample Output
3

思路

这是一道很有趣的题目,很容易想到符合题意得数就是素数的次幂,因为次幂至少从2开始,所以打一个1e6的素数表就可以了,那么我们可以在$[l_1,r_1]$和$[l_2,r_2]$里面枚举素数,因为重复的不算所以我们开始的时候要把范围有重叠的部分合并一下,然后再进行枚举,然后愉快的发现还是超时了,所以我们得优化,我们可以把不超过1e12的素数的次幂的表打出来,然后就可以不用每次重复计算次幂了,然后又愉快的tle了,然后我们才发现这题得用二分查找来优化,注意上下界的关系所以上界用upper_bound下界用lower_bound

#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cstring>
#include <algorithm>
#define MAX 1000005
using namespace std;
long long su[MAX],cnt;
bool isprime[MAX];
void prime()
{
    cnt=1;
    memset(isprime,1,sizeof(isprime));
    isprime[0]=isprime[1]=0;
    for(long long i=2;i<=MAX;i++)
    {
        if(isprime[i])
        {
            su[cnt++]=i;
        }
        for(long long j=i*2;j<=MAX;j+=i)
        {
            isprime[j]=0;
        }
    }
}
long long num[100000];
int n;
void init()
{
    n=0;
   for(int i=1;i<=cnt&&su[i]*su[i]<=1e12;i++)
            {
                long long x=su[i];
                long long y=su[i];
                while(1)
                {
                    x=x*y;
                    if(x<=1e12&&x>=1)
                        num[n++]=x;
                    if(x>1e12)
                        break;
                }
            }
}
int main()
{
    prime();
    init();
    sort(num,num+n);
    int t;
    scanf("%d",&t);
    while(t--)
    {
        long long l1,r1,l2,r2;
        scanf("%lld%lld%lld%lld",&l1,&r1,&l2,&r2);
        int flag=0;
        if(l1>=l2&&r1<=r2)
        {
            l1=l2;
            r1=r2;
            flag=1;
        }
        else if(l1<=l2&&r1<=r2&&r1>=l2)
        {
            r1=r2;
            flag=1;
        }
        else if(l1>=l2&&r1>=r2&&l1<=r2)
        {
            l1=l2;
            flag=1;
        }
        else if(l1<=l2&&r1>=r2)
        {
            flag=1;
        }
        if(flag==1)
        {
            int ans=0;
            ans=upper_bound(num,num+n,r1)-num;
            ans=ans-(lower_bound(num,num+n,l1)-num);
            printf("%d\n",ans);
        }
        else
        {
            int ans1;
            ans1=upper_bound(num,num+n,r1)-num;
            ans1=ans1-(lower_bound(num,num+n,l1)-num);
            int ans2;
            ans2=upper_bound(num,num+n,r2)-num;
            ans2=ans2-(lower_bound(num,num+n,l2)-num);
            printf("%d\n",ans1+ans2);
        }
    }
    return 0;
}