蓝桥杯2020决赛C++研究生组重复字符串

题目描述

如果一个字符串 SS 恰好可以由某个字符串重复 KK 次得到，我们就称 SS 是 KK 次重复字符串。例如 abcabcabc 可以看作是 abc 重复 33 次得到，所以 abcabcabc 是 33 次重复字符串。

同理 aaaaaa 既是 22 次重复字符串、又是 33 次重复字符串和 66 次重复字符串。

现在给定一个字符串 SS，请你计算最少要修改其中几个字符，可以使 SS 变为一个 KK 次字符串？

输入描述

输入第一行包含一个整数 KK。

第二行包含一个只含小写字母的字符串 SS。

其中，1 ≤ K ≤ 10^5, 1 ≤ |S| ≤ 10^51≤K≤105,1≤∣S∣≤105。其中 |S|∣S∣ 表示 SS 的 长度。

输出描述

输出一个整数代表答案。如果 SS 无法修改成 KK 次重复字符串，输出 −1−1。

 

思路：算出子字符串的长度，然后用一个二维数组记录每个位置字符出现的次数，同时记录每个位置字符数目出现最多的次数，然后累计计算即可，类似于分块思想。（当时我还在想怎么套DP，想了好久都没思路）

参考链接：重复字符串 - 蓝桥云课 (lanqiao.cn)的题解

#include<bits/stdc++.h>

using namespace std;
string S;
int K;

const int N = 1e5+10;
int maxn[N];
int num[N][26];

int calc(string& S, int len, int K)
{
    if( len % K != 0)
        return -1;
    if( K == len)
        return 0;

    memset(num,0,sizeof(num));
    memset(maxn,0,sizeof(maxn));

    int cnt = 0;
    int n = len / K;  //每一个子串的长度
    for(int i = 0; i < len; ++i)
    {
        ++num[i % n][S[i] -'a'] ;
        maxn[i%n] = max(maxn[i%n], num[i%n][S[i]-'a']);
    }

    for(int i = 0; i < n; ++i)
    {
        cnt += (K - maxn[i]);
    }
    return cnt;

}

int main()
{
    cin >> K;
    cin >> S;

    int len = S.size();
    int ans = calc(S, len, K);
    cout << ans << endl;

    return 0;
}