【PTA】C语言 航空公司VIP客户查询

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

输入格式:

输入首先给出两个正整数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

 用链表解决这道问题。

 

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>

typedef struct node {
    char id[20];
    int length;
    struct node* next;
} Node;

typedef struct {
    int size;
    Node* linklist;
} HashList;

int isPrime(int x) {
    int flag = 1;
    for (int i = 2; i <= (int)sqrt(x); i++) {
        if (x % i == 0) {
            flag = 0;
            break;
        }
    }
    return flag;
}

int nextprime(int n) {
    int i;
    for (i = n;; i++) {
        if (isPrime(i) == 1)
            break;
    }
    return i;
}

HashList* InitHash(int n) {
    HashList* hash = (HashList*)malloc(sizeof(HashList));
    hash->size = nextprime(n);
    hash->linklist = (Node*)malloc(sizeof(Node) * hash->size);
    for (int i = 0; i < hash->size; i++) {
        hash->linklist[i].id[0] = '\0';
        hash->linklist[i].length = 0;
        hash->linklist[i].next = NULL;
    }
    return hash;
}

int hashhash(HashList* hash, char id[]) {
    int index = 0;
    for (int i = 13; i <= 17; i++) {
        if (id[i] == 'x') {
            index = index * 10 + 10;
        }
        else {
            index = index * 10 + id[i] - '0';
        }
    }
    return index;
}

Node* find(HashList* hash, char id[]) {
    int index = hashhash(hash, id) % (hash->size);
    Node* ptr = hash->linklist[index].next;
    while (ptr && strcmp(ptr->id, id))
        ptr = ptr->next;
    return ptr;
}

void insert(HashList* hash, char id[], int length) {
    Node* p = find(hash, id);
    if (!p) {
        Node* newnode = (Node*)malloc(sizeof(Node));
        int index = hashhash(hash, id) % (hash->size);
        newnode->next = hash->linklist[index].next;
        hash->linklist[index].next = newnode;
        strcpy(newnode->id, id);
        newnode->length = length;
    }
    else {
        p->length = p->length + length;
    }
}

int main() {
    int N, K, M, length;
    HashList* hash;
    char id[20];
    scanf("%d %d", &N, &K);
    hash = InitHash(N);
    for (int i = 0; i < N; i++) {
        scanf("%s %d", id, &length);
        if (length < K)
            length = K;
        insert(hash, id, length);
    }
    scanf("%d", &M);
    for (int i = 0; i < M; i++) {
        scanf("%s", id);
        Node* p = find(hash, id);
        if (p == NULL)
            printf("No Info\n");
        else
            printf("%d\n", p->length);
    }

    return 0;
}