现在的位置: 首页 > 综合 > 正文

hdu 1800 Flying to the Mars(字典树)

2019年08月19日 ⁄ 综合 ⁄ 共 982字 ⁄ 字号 评论关闭

题目链接:hdu 1800 Flying to the Mars

题目大意:N个士兵,每个士兵有一个学习的能力,学习能力强的可以教学习能力弱的,现在N个士兵需要学习使用飞

行扫帚,每一个士兵在教下一个士兵的时候,会把自己练习的扫帚给他,问说N个士兵最少需要多少个扫帚。

解题思路:问题转换成求众数,因为士兵的等级是一个字符串,所以用字典树来统计,需要去除前导0。

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int maxn = 90005;
const int sigma_size = 10;

struct Tire {
    int sz, ans;
    int g[maxn][sigma_size];
    int val[maxn];

    void init();
    int idx(char ch);
    void insert(char* s);
    int find(char* s);
}T;

int main () {
    int n;
    char w[35];
    while (scanf("%d", &n) == 1) {
        T.init();
        for (int i = 0; i < n; i++) {
            scanf("%s", w);
            int v = 0;
            while (w[v] == '0') v++;
            T.insert(w + v);
        }
        printf("%d\n", T.ans);
    }
    return 0;
}

void Tire::init() {
    sz = 1;
    ans = val[0] = 0;
    memset(g[0], 0, sizeof(g[0]));
}

int Tire::idx (char ch) {
    return ch - '0';
}

int Tire::find(char* s) {
    int u = 0, n = strlen(s);

    for (int i = 0; i < n; i++) {
        int v = idx(s[i]);

        if (g[u][v] == 0)
            return 0;
        u = g[u][v];
    }
    return val[u];
}

void Tire::insert(char* s) {
    int u = 0, n = strlen(s);

    for (int i = 0; i < n; i++) {
        int v = idx(s[i]);

        if (g[u][v] == 0) {
            val[sz] = 0;
            memset(g[sz], 0, sizeof(g[sz]));
            g[u][v] = sz++;
        }

        u = g[u][v];
    }
    val[u]++;
    ans = max(ans, val[u]);
}

抱歉!评论已关闭.