7-45 航空公司VIP客户查询（PTA算法中文集）

7-45 航空公司VIP客户查询

分数 25

作者 DS课程组

单位 浙江大学

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

输入格式:

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

输出格式:

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

输入样例:

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

输出样例:

800
15000
1000
No Info

first:

         unordered_map:

#include <iostream>
#include <cstring>
#include <algorithm>
#include <unordered_map>

using namespace std;

unordered_map<string,int> mp;

int main()
{
    
    int n,t;
    scanf("%d%d",&n,&t);
    
    for(int i=0;i<n;++i)
    {
        string id;
        int num;
        id.resize(18);
        scanf("%s%d",&id[0],&num);
        
        if(num < t)
            mp[id] += t;
        else
            mp[id] += num;
    }
    
    int m;
    scanf("%d",&m);
    
    for(int i=0;i<m;++i)
    {
        string id;
        id.resize(18);
        scanf("%s",&id[0]);
        if(mp[id] == 0)
            puts("No Info");
        else
            printf("%d\n",mp[id]);
    }
    return 0;
}

运行结果：

second:

        Tire树，字典树：

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cmath>

using namespace std;

const int maxn = 1e6;
int son[maxn][11],cnt[maxn];
int idx;

int t;

void insert(const char *id,int dis)
{
    int p = 0;
    for(int i=0;i<18;++i)
    {
        int u;
        if(isalpha(id[i]))
            u = 10;
        else
            u = id[i] - '0';
        if(son[p][u] == 0)
            son[p][u] = ++idx;
        p = son[p][u];
    }
    if(dis < t)
        cnt[p] += t;
    else
        cnt[p] += dis;
}

int query(const char *id)
{
    int p = 0;
    for(int i=0;i<18;++i)
    {
        int u;
        if(isalpha(id[i]))
            u = 10;
        else
            u = id[i] - '0';
        if(son[p][u] == 0)
            return 0;
        p = son[p][u];
    }
    return cnt[p];
}

int main()
{
    int n;
    scanf("%d%d",&n,&t);
    
    char id[18];
    int dis;
    for(int i=0;i<n;++i)
    {
        scanf("%s%d",id,&dis);
        insert(id,dis);
    }
    
    int m;
    scanf("%d",&m);
    
    for(int i=0;i<m;++i)
    {
        scanf("%s",id);
        int res = query(id);
        if(res == 0)
            puts("No Info");
        else
            printf("%d\n",res);
    }
    
    return 0;
    
}

运行结果：

 对比来看，还是unordered_map 更胜一筹！

字典树学习传送门：

835. Trie字符串统计，836。最大异或对，(Tire树，字典树)_疯疯癫癫才自由的博客-CSDN博客