每日OJ题_牛客_数组中两个字符串的最小距离

目录

牛客_数组中两个字符串的最小距离

解析代码

---

牛客_数组中两个字符串的最小距离

数组中两个字符串的最小距离__牛客网

        给定一个字符串数组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;
}