题目来源:https://leetcode-cn.com/problems/repeated-string-match/
大致题意:
给一个字符串 a 和字符串 b,求出 a 最少复制几次后 b 才可以成为 a 复制后的子串
思路
如果想要 b 成为 a 复制后的子串,那么复制后 a 的长度必须大于等于 b 的长度
那么会有以下三种情况:
- 当 a 复制 n 次后,正好长度和 b 一致,且 a 与 b 相同,那么 n 就是所求的次数
- 当 a 复制 n + 1 次后,b 是复制后的字符串的子串,此时 a 的长度大于 b
- 当 a 复制 n + 1 次后,b 不是复制后的字符串的子串
于是解题方法可以为:
- 将 a 复制 n+1 次,并存下次数 n
- 判断 b 是不是当前的复制后字符串的子串,若不是直接返回 -1
- 若是,那么判断 b 在复制后字符串中的结尾位置是否大于 n * a 的原长度 ,如果大于,表示需要复制 n+1 次才满足题意,返回 n+1;否则,返回 n
代码:
public int repeatedStringMatch(String a, String b) {
StringBuffer sb = new StringBuffer();
int ans = 0;
// 先复制 n 次
while (sb.length() < b.length()) {
ans++;
sb.append(a);
}
// 再复制一次,共 n+1 次
sb.append(a);
// 判断 b 是不是复制后字符串的子串
int idx = sb.indexOf(b);
// 如果不是直接返回
if (idx == -1) {
return -1;
}
// 判断 b 在复制后字符串出现的结尾位置 idx + b.length()
// 与复制 n 次的字符串长度大小,如果大于那么表示需要复制 n+1 次
return idx + b.length() > ans * a.length() ? ans + 1 : ans;
}