码蹄集2198--个数统计2

sweet_Mary

已于 2023-09-06 21:53:01 修改

阅读量99

点赞数 1

分类专栏：码蹄集文章标签：算法数据结构 c++

于 2023-09-05 18:52:47 首次发布

本文链接：https://blog.csdn.net/sweet_Mary/article/details/132698253

版权

码蹄集专栏收录该内容

21 篇文章 1 订阅

订阅专栏

本文介绍了如何使用高精度计算配合Knuth-Morris-Pratt(KMP)算法解决字符串构造问题，包括KMP求解模板、高精度乘法处理以及字符串匹配计数。

摘要由CSDN通过智能技术生成

高精度*低精度+KMP求解

高精度*低精度：码蹄集2197--个数统计_sweet_Mary的博客-CSDN博客

KMP：码蹄集2176--字符串构造_sweet_Mary的博客-CSDN博客（这里只写了next数组的求解模板）

这里补上KMP的求解模板

int kmp(string a,int len_a, string b,int len_b) {
    int i = 0, j = 0;
    int temp = 0;
    
    while (i < len_a && j<len_b) {
        if (j == -1 || a[i]==b[i]) {
            i++; j++;
        }
        else {
            j = next_[j];
        }
    }
    if(j==len_b){
        return i-j;
    }
    else{return -1;}
}

本题的代码：

//高精度计算+KMP
#include <string>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>

using namespace std;
int N;
string b;
//a*新来的数放入res中
int a[3010];
int res[3010];
int cnt[3010];
int next_[3010];

void high_multiple(int num, int& len) {

    for (int i = 0; i < len; i++) {
        res[i] += a[i] * num;
        if (res[i] >= 10) {
            res[i + 1] += res[i] / 10;
            res[i] %= 10;
        }
    }
    while (res[len]) {
        if (res[len] >= 10) {
            res[len + 1] += res[len] / 10;
            res[len] %= 10;
        }
        len++;
    }

    while (res[len - 1] == 0 && len > 1) {
        len--;
    }
   
}

void find_next(int len_b) {
    int j = -1, i = 0;
    next_[0] = -1;
    while (i <= len_b) {
        if (j == -1 || b[i] == b[j]) {
            i++;
            j++;
            next_[i] = j;
        }
        else {
            j = next_[j];
        }
    }
}

int kmp(int len, int len_b) {//int类型的cnt数组为文本串，string类型的b是模式串
    int i = 0, j = 0;
    int temp = 0;
    

    while (i < len) {
      
        if (j == -1 || cnt[i] == (b[j] - '0')) {
            i++; j++;
        }
        else {
            j = next_[j];
        }
        if (j == len_b) {
            temp++;
            j = next_[j];
        }
    }
    return temp;
}

int main()
{
    cin >> N >> b;
    int len_b = b.size();
    int len = 1;
    a[0] = 1;
    for (int i = 1; i <= N; i++) {
        high_multiple(i, len);
        memcpy(a, res, sizeof(res));
        memset(res, 0, sizeof(res));
    }
    //将结果逆序存放在a数组中
    //将结果转变到cnt数组中
    for (int i = 0; i < len; i++) {
        cnt[i] = a[len - i - 1];
    }
    find_next(len_b);

    cout<<kmp(len, len_b);
    return 0;
}