作者推荐
视频算法专题
本文涉及知识点
动态规划汇总
二分查找
力扣:466 统计重复个数
定义 str = [s, n] 表示 str 由 n 个字符串 s 连接构成。
例如，str == [“abc”, 3] ==“abcabcabc” 。
如果可以从 s2 中删除某些字符使其变为 s1，则称字符串 s1 可以从字符串 s2 获得。
例如，根据定义，s1 = “abc” 可以从 s2 = “abdbec” 获得，仅需要删除加粗且用斜体标识的字符。
现在给你两个字符串 s1 和 s2 和两个整数 n1 和 n2 。由此构造得到两个字符串，其中 str1 = [s1, n1]、str2 = [s2, n2] 。
请你找出一个最大整数 m ，以满足 str = [str2, m] 可以从 str1 获得。
示例 1：
输入：s1 = “acb”, n1 = 4, s2 = “ab”, n2 = 2
输出：2
示例 2：
输入：s1 = “acb”, n1 = 1, s2 = “acb”, n2 = 1
输出：1
提示：
1 <= s1.length, s2.length <= 100
s1 和 s2 由小写英文字母组成
1 <= n1, n2 <= 106
动态规划
inxs[i] 记录s1中’a’+i的下标，升序。
分两步：
一，动态规划求dp[i]。dp[i]的含义是:s1[i,m_c1)+s1+s1… 包括s2的最短前缀长度。
dp[i]独立计算：
通过ch遍历s2， ii是s1对应的下标。如果s1不存在ch，则直接返回0。
如果inxs[ch-‘a’]存在大于等于ii的下标ij，则ch和ij对应。ii=ij+1。
如果不存在 ，iTurn ++ ii = inxs[ch-‘a’].front()+1
dp[i] = m_c1*iTurn + ii - i ;
这一步时间复杂度:O(nnlogn)
二，循环i，看[s1,n1]能否包括i个s2。
极端情况，时间复杂度O(108)
s1是100个a，n1是106。s2是’a’。
代码
核心代码
class Solution {
public:
int getMaxRepetitions(string s1, int n1, string s2, int n2) {
m_c1 = s1.length();
m_c2 = s2.length();
vector<int> inxs[26];
for (int i = 0; i < m_c1; i++)
{
inxs[s1[i] - 'a'].emplace_back(i);
}
vector<int> dp(m_c1);//dp[i]的含义是:s1[i,m_c1)+s1+s1.... 包括s2的最短前缀长度
for (int i = 0; i < m_c1; i++)
{
int ii = i;
int iTrun = 0;
for (int j = 0; j < m_c2; j++)
{
const auto& inx = inxs[s2[j] - 'a'];
if (inx.empty())
{//某个字符不存在
return 0;
}
const auto it = std::lower_bound(inx.begin(), inx.end(), ii);
if (inx.end() != it )
{
ii = *it+1;
}
else
{
ii = inx.front() + 1;
iTrun++;
}
}
dp[i] = m_c1 * iTrun + ii-i;
}
const int iHas = m_c1 * n1;
int iNeed = 0;
int inx = 0;
for (int i = 0; ; i++)
{
iNeed += dp[inx];
inx = (inx + dp[inx]) % m_c1;
if (iNeed > iHas)
{
return i/n2;//i最多包括多少个s2
}
}
return 0;
}
int m_c1,m_c2;
};
测试用例
template<class T>
void Assert(const T& t1, const T& t2)
{
assert(t1 == t2);
}
template<class T>
void Assert(const vector<T>& v1, const vector<T>& v2)
{
if (v1.size() != v2.size())
{
assert(false);
return;
}
for (int i = 0; i < v1.size(); i++)
{
Assert(v1[i], v2[i]);
}
}
int main()
{
string s1, s2;
int n1, n2;
{
Solution sln;
s1 = "acb", n1 = 4, s2 = "ab", n2 = 2;
auto res = sln.getMaxRepetitions(s1, n1, s2, n2);
Assert(2, res);
}
{
Solution sln;
s1 = "acb", n1 = 1, s2 = "acb", n2 = 1;
auto res = sln.getMaxRepetitions(s1, n2, s2, n2);
Assert(1, res);
}
}
优化
inx 取值范围[0,m_c1)，所以m_c1+1次必定重复。 重复的部分只计算一次。
vBuf[inx]，消耗了first个s2 时，s1消耗了second个字符 ，即s1消耗了second/m_c1个 ，还消耗了s1[0,second%m_c1)。
class Solution {
public:
int getMaxRepetitions(string s1, int n1, string s2, int n2) {
m_c1 = s1.length();
m_c2 = s2.length();
vector<int> inxs[26];
for (int i = 0; i < m_c1; i++)
{
inxs[s1[i] - 'a'].emplace_back(i);
}
vector<int> dp(m_c1);//dp[i]的含义是:s1[i,m_c1)+s1+s1.... 包括s2的最短前缀长度
for (int i = 0; i < m_c1; i++)
{
int ii = i;
int iTrun = 0;
for (int j = 0; j < m_c2; j++)
{
const auto& inx = inxs[s2[j] - 'a'];
if (inx.empty())
{//某个字符不存在
return 0;
}
const auto it = std::lower_bound(inx.begin(), inx.end(), ii);
if (inx.end() != it )
{
ii = *it+1;
}
else
{
ii = inx.front() + 1;
iTrun++;
}
}
dp[i] = m_c1 * iTrun + ii-i;
}
const int iTotal = m_c1 * n1;
int iHas = 0;
int inx = 0;
vector<pair<int,int>> vBuf(m_c1,std::pair<int,int>(-1,-1));//消耗了first个s2 时，s1消耗了second个字符
vBuf[0] = std::pair<int, int>(0, iHas);
for (int i = 0; ; i++)
{
iHas += dp[inx];
inx = (inx + dp[inx]) % m_c1;
if (iHas > iTotal)
{
return i/n2;//i最多包括多少个s2
}
if (-1 == vBuf[inx].first)
{
vBuf[inx] = std::make_pair(i+1, iHas);
}
else
{
const int subHas =iHas - vBuf[inx].second ;
const int subI = (i+1) - vBuf[inx].first;
i += (iTotal-iHas) / subHas * subI;
iHas += (iTotal - iHas) / subHas*subHas;
}
}
return 0;
}
int m_c1,m_c2;
};
再次优化：动态规划
dp[i][j]的含义是：s1[i,m_c1)+s1+s1… 包括s2[j,m_c2)的最短前缀长度
class Solution {
public:
int getMaxRepetitions(string s1, int n1, string s2, int n2) {
m_c1 = s1.length();
m_c2 = s2.length();
vector<vector<int>> dp(m_c1, vector<int>(m_c2));//dp[i][j]的含义是：s1[i,m_c1)+s1+s1.... 包括s2[j,m_c2)的最短前缀长度
for (int j = m_c2 - 1; j >= 0; j--)
{
vector<int> inxs;
for (int i = 0; i < m_c1; i++)
{
if (s2[j] == s1[i])
{
inxs.emplace_back(i);
}
}
if (inxs.empty())
{
return 0;
}
for (int i = 0, k = 0; i < m_c1; i++)
{
while ((k < inxs.size()) && (inxs[k] < i))
{
k++;
}
dp[i][j] = (inxs.size() == k) ? (m_c1 - i + inxs.front() + 1) : (inxs[k] - i + 1);
if (m_c2 - 1 == j)
{
continue;
}
const int inx = (i + dp[i][j]) % m_c1;
dp[i][j] += dp[inx][j + 1];
}
}
const int iTotal = m_c1 * n1;
int iHas = 0;
int inx = 0;
vector<pair<int, int>> vBuf(m_c1, std::pair<int, int>(-1, -1));//消耗了first个s2 时，s1消耗了second个字符
vBuf[0] = std::pair<int, int>(0, iHas);
for (int i = 0; ; i++)
{
iHas += dp[inx][0];
inx = (inx + dp[inx][0]) % m_c1;
if (iHas > iTotal)
{
return i / n2;//i最多包括多少个s2
}
if (-1 == vBuf[inx].first)
{
vBuf[inx] = std::make_pair(i + 1, iHas);
}
else
{
const int subHas = iHas - vBuf[inx].second;
const int subI = (i + 1) - vBuf[inx].first;
i += (iTotal - iHas) / subHas * subI;
iHas += (iTotal - iHas) / subHas * subHas;
}
}
return 0;
}
int m_c1, m_c2;
};
2023年1月 版
class Solution {
public:
int getMaxRepetitions(string s1, int n1, string s2, int n2) {
int iCnt1 = 0, iCnt2 = 0, index = 0;
std::unordered_map<int, pair<int, int>> mIndexToCnt;
while (true)
{
for (const auto& ch : s1)
{
if (ch == s2[index])
{
index++;
if (s2.length() == index)
{
index = 0;
iCnt2++;
}
}
}
iCnt1++;
if (iCnt1 == n1)
{//已经匹配完毕
return iCnt2 / n2;
}
if (mIndexToCnt.count(index))
{//找到循环节
break;
}
else
{
mIndexToCnt[index] = { iCnt1, iCnt2 };
}
}
int iLoopCnt1 = iCnt1 - mIndexToCnt[index].first;
int iLoopCnt2 = iCnt2 - mIndexToCnt[index].second;
int iRet = iCnt2 + (n1 - iCnt1) / iLoopCnt1 * iLoopCnt2;
int iRemain = (n1 - iCnt1) % iLoopCnt1;
for (int i = 0; i < iRemain; i++)
{
for (const auto& ch : s1)
{
if (ch == s2[index])
{
index++;
if (s2.length() == index)
{
index = 0;
iRet++;
}
}
}
}
return iRet/n2;
}
};
