HDU 1800 Flying to the Mars // Trie

题目描述

HDU 1800 Flying to the Mars

解题思路

题目大意：
每个士兵都有各自的能力值，能力强的可以教能力弱的人 “骑扫帚”，两人用的是同一把扫帚 (若A B 用 一把， B C 用一把， 则说明 A B C用同一把）。另外题目要求每个士兵最多只能有一个学生或最多只能有一个老师。最后输出最少需要多少扫帚~

不难知道，若n的士兵能力值各不相同的话，则他们只需要一把扫帚。因此，最终扫帚的数目等于 有相同能力值的人数中 取最大的那个即可。
比如能力值 1 2 3 4 5 只要1把； 而 2 2 5 5 5 则需要 3把。

另外还需注意的是，能力值的输入是有前导“0”的，即003 和 3 应认为是相同的。

参考代码

//**********************************************
//  Author: @xmzyt1996
//  Date:   2015-10-21
//  Name:   HDU 1800.cpp
//**********************************************
#include <cstdio>
#include <cmath>
#include <cctype>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#include <string>
#include <bitset>
#include <vector>
#include <stack>
#include <queue>
#include <map>
using namespace std;
#define clr(a, b) memset(a, b, sizeof(a))
#define rep(i, a, b) for(int i = a; i < b; ++i)
#define per(i, a, b) for(int i = a; i >= b; --i)
#define pt(x) cout << #x << " = " << x << endl
#define ps puts("debug~~")
#define all(x) (x).begin(),(x).end()
#define mp make_pair
#define pb push_back
typedef __int64 ll;
typedef pair<int, int> pii;
const double pi = acos(-1.0);
const double eps = 1e-6;
const int inf = 0x3f3f3f3f;
const int MOD = 1e9+7;
const int MAX_N = 3010;

struct Trie {
    Trie* next[10];
    int cnt;
} node[MAX_N];

int num, ans;
char s[35];

void init () {
    num = ans = 0;
    clr(node, 0);
}

int getid(char ch) { return ch - '0'; }

char* check (char* s) { // 除掉前导“0”
    int i = 0;
    for (; s[i+1] && s[i] == '0'; ++i);
    return s+i;
}

void insert (char* s) {
    Trie* head = &node[0];
    while (*s) {
        int id = getid(*s++);
        if (head->next[id] == NULL)
            head->next[id] = &node[++num];
        head = head->next[id];
    }
    (head->cnt)++;
    if ((head->cnt) > ans) ans = (head->cnt);
}

int main() {
    int n;
    while (~scanf("%d", &n)) {
        init();
        rep (i, 0, n) {
            clr(s, 0);
            scanf("%s", s);
            insert(check(s));
        }
        printf("%d\n", ans);
    }
    return 0;
}