Codeforces Round #226 (Div. 2) C. Bear and Prime Numbers【素數篩】

題目鏈接:https://codeforc.es/problemset/problem/385/C
題目大意:給定nn個數,定義f(p)f(p)nn個數中爲pp的倍數的個數,qq次查詢區間[l,r][l,r],計算pf(p)\sum_{p}f(p)pp爲區間中的素數。
思路:由於nn的數的值域不大,我們可以直接先記錄每個數出現的次數,然後素數篩的時候計算素數pp及其倍數出現的總次數,最後在處理一下前綴和就可以每次O(1)O(1)查詢區間了,區間超過1e71e7的無效,注意越界問題。
AC代碼:

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e7+10;
int sum[maxn];
int vis[maxn];
int ss[maxn];
int main()
{
    int n;
    scanf("%d",&n);
    int t;
    for(int i=1;i<=n;i++){
        scanf("%d",&t);
        sum[t]++;
    }
    for(int i=2;i<maxn;i++){
        if(vis[i]==0){
            for(int j=2*i;j<maxn;j+=i){
                sum[i]+=sum[j];
                vis[j]=1;
            }
        }
    }
    ss[0]=0;
    for(int i=1;i<maxn;i++){
        ss[i]=ss[i-1];
        if(vis[i]==0){
            ss[i]+=sum[i];
        }
    }
    int m,l,r;
    scanf("%d",&m);
    while(m--){
        scanf("%d%d",&l,&r);
        l=min(maxn-10,l);
        r=min(r,maxn-10);
        printf("%d\n",ss[r]-ss[l-1]);
    }
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章