T216909 小卡与质数2(字典树）

题目背景

小卡迷上了质数！

题目描述

小卡最近迷上了质数，所以他想把任何一个数都转化为质数！

小卡有 TT 次询问，每次给你一个数字 xx，问有多少个比 xx 小的非负整数 yy，使得 x\oplus yx⊕y 是质数，其中 \oplus⊕ 表示按位异或。

输入格式

第一行一个正整数 T(1\le T\le10^5)T(1≤T≤105)，表示有 TT 组询问。

接下来 TT 行，每行一个正整数 x(1\le x\le 10^6)x(1≤x≤106)。

输出格式

对于每组询问，输出一行一个整数，表示答案。

输入输出样例

输入 #1复制

9
5
6
7
8
9
10
100
1000
10000

输出 #1复制

2
4
4
2
2
4
22
163
1132

 字典树：

#include<bits/stdc++.h>
using namespace std;
const int MAXN = 2.1e6 + 10;

int T, x;
int ans[MAXN];
int prim[MAXN], Cnt = 0;
bool vis[MAXN], flag[MAXN];

void Init()
{
    int M = 2100000;
    for(int i = 2; i <= M; ++i)
    {
        if(!vis[i]) prim[++Cnt] = i;
        for(int j = 1; j <= Cnt && i * prim[j] <= M; ++j)
        {
            vis[i * prim[j]] = true;
            if(i % prim[j] == 0) break;
        }
    }
}

namespace Trie
{
int tr[MAXN << 2][2], node_num = 0, siz[MAXN << 2];
void Insert(int x)
{
    int u = 0;
    siz[0] ++;
    for(int j = 21; j >= 0; --j)
    {
        int c = (x >> j) & 1;
        if(!tr[u][c]) tr[u][c] = ++node_num;
        u = tr[u][c];
        siz[u] ++;
    }
}
int Query(int x)
{
    int u = 0, res = 0;
    for(int j = 21; j >= 0; --j)
    {
        int c = (x >> j) & 1;
        if(c)
        {
            res = res + siz[tr[u][c]];
            u = tr[u][c ^ 1];
        }
        else
        {
            u = tr[u][c];
        }
        if(!u) return res;
    }
    return res;
}
}

int main()
{
    Init();
    for(int i = 1; i <= Cnt; ++i)
    {
        Trie::Insert(prim[i]);
    }
    scanf("%d", &T);
    while(T--)
    {
        scanf("%d", &x);
        int res = Trie::Query(x);
        printf("%d\n", res);
    }
    return 0;
}

本质：

#include<bits/stdc++.h>
using namespace std;
const int MAXN = 2.1e6 + 10;

int T, x;
int ans[MAXN], a[200];
int prim[MAXN], Cnt = 0;
bool vis[MAXN], flag[MAXN];

int read() {
    int s = 0, f = 0;
    char ch = getchar();
    while(!isdigit(ch)) f |= (ch == '-'), ch = getchar();
    while(isdigit(ch)) s = (s << 1) + (s << 3) + (ch ^ 48), ch = getchar();
    return f ? -s : s;
}

void Init() {
    int M = 2100000;
    for(int i = 2; i <= M; ++i) {
        if(!vis[i]) prim[++Cnt] = i;
        for(int j = 1; j <= Cnt && i * prim[j] <= M; ++j) {
            vis[i * prim[j]] = true;
            if(i % prim[j] == 0) break;
        }
    }
}

int main() {
    Init();
    for(int i = 1; i <= Cnt; ++i) {
        int x = prim[i];
        int lst, cnt = 0;
        while(x) {
            if(x & 1) lst = cnt;
            x >>= 1, ++ cnt;
        }
        a[lst] ++;
    }
    cin >> T;
    while(T--) {
        int Ans = 0;
        cin >> x;
        for(int i = 0; i <= 21; ++i) 
            if(x & (1 << i)) Ans += a[i];
        printf("%d\n", Ans);
    }
    return 0;
}