航空公司VIP客户查询 （25 分)

不少航空公司都会提供优惠的会员服务，当某顾客飞行里程累积达到一定数量后，可以使用里程积分直接兑换奖励机票或奖励升舱等服务。现给定某航空公司全体会员的飞行记录，要求实现根据身份证号码快速查询会员里程积分的功能。

输入格式:
输入首先给出两个正整数N（≤10
​5
​​ ）和K（≤500）。其中K是最低里程，即为照顾乘坐短程航班的会员,航空公司还会将航程低于K公里的航班也按K公里累积。随后N行，每行给出一条飞行记录。飞行记录的输入格式为：18位身份证号码（空格）飞行里程。其中身份证号码由17位数字加最后一位校验码组成，校验码的取值范围为0~9和x共11个符号；飞行里程单位为公里，是(0, 15 000]区间内的整数。然后给出一个正整数M（≤10
​5
​​ ），随后给出M行查询人的身份证号码。

输出格式:
对每个查询人，给出其当前的里程累积值。如果该人不是会员，则输出No Info。每个查询结果占一行。

输入样例:

4 500
330106199010080419 499
110108198403100012 15000
120104195510156021 800
330106199010080419 1
4
120104195510156021
110108198403100012
330106199010080419
33010619901008041x

输出样例:

800
15000
1000
No Info

第一次用的string加unordered_map超时了，代码可以看一看

#include <cstdio>
#include <cstdlib>
#include <iostream>
#include <string>
#include <unordered_map>
using namespace std;
int n,m,v,cnt,k,a;
unordered_map<string,int> dp;
string id;
int main(){
    cin>>n>>k;
    for(int i=0;i<n;i++){
        cin>>id>>a;
        dp[id]+=a<k?k:a;
    }
    cin>>n;
    for(int i=0;i<n;i++){
        cin>>id;
        if(dp.find(id)==dp.end()) printf("No Info\n");
        else printf("%d\n",dp[id]);
    }
    return 0;
}

既然超时，就想到了前缀树，就用了一个一般的前缀树实现，用的结构体实现，结果超内存了，代码如下

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <string>
using namespace std;
int n,m,v,x,k,a;
struct trie{
    trie* next[11];
    int grade=0;
    trie(){
        memset(next,0,sizeof(trie*)*11);
    }
};
int mp(char a){
    return isdigit(a)?(a-'0'):10;
}
char path[19];
int main(){
    scanf("%d %d\n",&n,&k);
    trie* root=new trie();
    int step;
    trie* temp;
    int rel;
    for(int i=0;i<n;i++){
        scanf("%s %d\n",path,&a);
        step=0;
        temp=root;
        while(step!=18){
            rel=mp(path[step]);
            if(!temp->next[rel]) temp->next[rel]=new trie();
            temp=temp->next[rel];
            step++;
            if(step==18) temp->grade+=a<k?k:a;
        }
    }
    scanf("%d\n",&n);
    for(int i=0;i<n;i++){
        scanf("%[^\n]\n",path);
        step=0;
        temp=root;  
        while(step!=18){
            rel=mp(path[step]);
            if(!temp->next[rel]){
                printf("No Info\n");
                break;
            }
            temp=temp->next[rel];
            step++;
            if(step==18) printf("%d\n",temp->grade);
        }

    }
    return 0;
}

最后只能出终极大招了，用数组模拟前缀树，代码如下，然后内存用了一半，时间用了一半终于过了，所以以后实现前缀树，直接用数组实现，更简单。。。

#include <cstdio>
#include <cstdlib>
#include <cstring>
using namespace std;
int n,m,v,x,k,a,cnt=2;
#define N 1000005
int trie[N][12];
int mp(char a){
    return isdigit(a)?(a-'0'):10;
}
char path[19];
int main(){
    scanf("%d %d\n",&n,&k);
    int root=1;
    int step;
    int rel;
    for(int i=0;i<n;i++){
        scanf("%s %d\n",path,&a);
        step=0;
        root=1;
        while(step!=18){
            rel=mp(path[step]);
            if(!trie[root][rel]) trie[root][rel]=cnt++;
            root=trie[root][rel];
            step++;
            if(step==18) trie[root][11]+=a<k?k:a;
        }
    }
    scanf("%d\n",&n);
    for(int i=0;i<n;i++){
        scanf("%[^\n]\n",path);
        step=0;
        root=1;
        while(step!=18){
            rel=mp(path[step]);
            if(!trie[root][rel]){
                printf("No Info\n");
                break;
            }
            root=trie[root][rel];
            step++;
            if(step==18) printf("%d\n",trie[root][11]);
        }
    }
    return 0;
}