[HDOJ1381]Crazy Search

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1381

Crazy Search

Time Limit: 10000/5000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 2236    Accepted Submission(s): 827

Problem Description

Many people like to solve hard puzzles some of which may lead them to madness. One such puzzle could be finding a hidden prime number in a given text. Such number could be the number of different substrings of a given size that exist in the text. As you soon will discover, you really need the help of a computer and a good algorithm to solve such a puzzle.

Your task is to write a program that given the size, N, of the substring, the number of different characters that may occur in the text, NC, and the text itself, determines the number of different substrings of size N that appear in the text.

As an example, consider N=3, NC=4 and the text "daababac". The different substrings of size 3 that can be found in this text are: "daa", "aab", "aba", "bab", "bac". Therefore, the answer should be 5.

 

Input

The first line of input consists of two numbers, N and NC, separated by exactly one space. This is followed by the text where the search takes place. You may assume that the maximum number of substrings formed by the possible set of characters does not exceed 16 Millions.

 

Output

The program should output just an integer corresponding to the number of different substrings of size N found in the given text.
The first line of a multiple input is an integer N, then a blank line followed by N input blocks. Each input block is in the format indicated in the problem description. There is a blank line between input blocks.

The output format consists of N output blocks. There is a blank line between output blocks.

 

Sample Input

1 3 4 daababac

 

Sample Output

5

 

hash，给每一个字符映射成一位nc进制的数，再给每一个子串算一个value（秦九韶算法），检查是否出现过，出现过就不统计了。貌似STL可以直接做

 

 

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <cmath>
#include <queue>
#include <map>
#include <stack>
#include <list>
#include <vector>

using namespace std;

const int maxn = 16000010;

bool has[maxn];
char str[maxn];
int ascii[256];
int n, nc;

int solve() {
    memset(has, 0, sizeof(has));
    memset(ascii, 0, sizeof(ascii));
    int cnt = 1;
    for(int i = 0; str[i]; i++) {
        if(!ascii[str[i]]) {
            ascii[str[i]] = cnt;
            cnt++;
        }
        if(cnt == nc) {
            break;
        }
    }
    int len = strlen(str) - n;
    int ans = 0;
    for(int i = 0; i <= len; i++) {
        int val = 0;
        for(int j = 0; j < n; j++) {
            val = val * nc + ascii[str[i+j]];
        }
        if(!has[val]) {
            has[val] = 1;
            ans++;
        }
    }
    return ans;
}

int main() {
    int T;
    scanf("%d", &T);
    while(T--) {
        scanf("%d %d", &n, &nc);
        getchar();
        scanf("%s", str);
        printf("%d
", solve());
    }
    return 0;
}