目录
牛客_数组中两个字符串的最小距离
给定一个字符串数组strs,再给定两个字符串str1和str2,返回在strs中str1和str2的最小距离,如果str1或str2为null,或不在strs中,返回-1。
解析代码
思路:贪心
1.如果暴力:两层for循环枚举,数据范围是10^5,会超时。
2.用贪心(dp)来优化。
3.用若干个变量来标记某个位置前驱的最优解(常用优化方法)预处理思想,只管左边,不管右边。当遍历到后面的时候,向左看,会包含之前右边的情况。
4.prev1:标记的是遍历到i位置时,左边最近符合要求的下标。
5.prev2:标记的是i遍历到了符合要求的另一个下标。
时间复杂度:O(N)
#include <iostream>
#include <unordered_map>
#include <vector>
#include <climits>
using namespace std;
int main()
{
string str1, str2;
int n = 0; // strs的长度
cin >> n >> str1 >> str2;
vector<string> strs(n);
for (int i = 0; i < n; ++i)
{
cin >> strs[i];
}
if (str1.empty() || str2.empty())
{
cout << -1;
return 0;
}
bool flag = false;
vector<int> leftPos, rightPos;
for (int i = 0; i < n; ++i)
{
if (strs[i] == str1)
{
leftPos.push_back(i);
}
else if (strs[i] == str2)
{
rightPos.push_back(i);
}
}
if (leftPos.empty() || rightPos.empty())
{
cout << -1;
return 0;
}
int res = INT_MAX;
for (int i = 0; i < leftPos.size(); ++i)
{
for (int j = 0; j < rightPos.size(); ++j)
{
res = min(res, abs(leftPos[i] - rightPos[j]));
}
}
cout << res;
return 0;
//for (int left = 0, right = 0; right < n; ++right)
//{
// string tmp; // 前一次出现的
// if (strs[right] == str1)
// {
// tmp = strs[right];
// begin = right;
// }
// else if (strs[right] == str1 || strs[right] == str2 && begin != -1)
// {
// tmp = strs[right];
// end = right;
// }
// while (tmp == strs[left])
//}
//return 0;
}