【C++】马蹄集06 字符串构造

你有一个字符串t，它由几个字母组成。定义一个字符串s的子串为s[l...r]，表示从位置l到r构成的个新的串。
你的目标是构造一个字符串s，使得它的可能长度最小，要求s中存在k个位置，可以找到k个以i为出发点的子串t。

格式

输入格式：第一行输入两个整数几和k，表示t的长度和需要k个子串（1<=n,k<=1e5，保证答案字符串的长度在3e5范围内）。第二行输入字符串t。
输出格式：输出满足条件的长度最小的s；
题目保证答案唯一。

样例1

 输入：3 4
aba
输出：ababababa 

代码  计算公共前后缀

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
using namespace std;

int g(const std::string& s, const std::string& t) {
    int lenS = s.length();
    int lenT = t.length();
    std::vector<int> commonLengths;

    // 检查所有可能的公共前后缀长度
    for (int i = 1; i <= std::min(lenS, lenT); ++i) {
        // 检查s的后缀和t的前缀是否匹配
        if (s.substr(lenS - i) == t.substr(0, i)) {
            commonLengths.push_back(i);
        }
    }

    // 根据条件选择最长的公共前后缀长度
    if (commonLengths.empty()) {
        return 0; // 没有公共前后缀
    }

    // 排序并选择第二长的公共前后缀长度
    std::sort(commonLengths.begin(), commonLengths.end(), std::greater<int>());
    int longest = commonLengths[0];
    if ((longest == lenS && longest == lenT) || longest == lenT) {
        return commonLengths.size() > 1 ? commonLengths[1] : 0;
    }

    return longest;
}

string f(int n, string t, int k) {
    string s = t;
    for (int i = 1; i < k; ++i) {
        // 寻找s末尾与t开始部分的最大共享长度
        int shared_length = g(s,t);

        // 更新s，添加t剩余不共享的部分
        if (shared_length > 0 && shared_length < n) {
            // 如果有共享部分，则只添加t中未共享的部分
            s += t.substr(shared_length);
        } else if(shared_length == 0){
            // 如果没有共享部分，添加整个t
            s += t;
        }
    }
    return s;
}

int main() {
    int n, k;
    cin >> n >> k;
    string t;
    cin >> t;

    string s = f(n, t, k);
    cout << s << endl;
    return 0;
}