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

给定一个字符串数组strs，再给定两个字符串str1和str2，返回在strs中str1和str2的最小距离，如果str1或str2为null，或不在strs中，返回-1。

输入描述：

输入包含有多行，第一输入一个整数n(1 ≤ n ≤ 10^5)，代表数组strs的长度，第二行有两个字符串分别代表str1和str2，接下来n行，每行一个字符串，代表数组strs(保证题目中出现的所有字符串长度均小于等于10)

输出描述：

输出一行，包含一个整数，代表返回的值。

补充说明：

时间复杂度O（n），额外空间复杂度O（1）

解法一：暴力解法（这里不举例）

解法二：贪心（dp）

prev 存放最近的数值下标

示例1：

输入：1
​ CD AB
​ CD
输出： -1

示例2：

输入：5
​ QWER 666
​ QWER
​ 1234
​ qwe
​ 666
​ QWER
输出： 1

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

int main()
{
    int n;
    int prev1 = -1, prev2 = -1,ret = -1;
    string s1, s2;
    string s;
    cin >>n;
    cin >>s1 >> s2;
    for (int i = 0; i < n; i++)
    {
        cin >> s;
        if(s == s1)  //去前面找最近的s2
        {
            if(prev2 != -1)
            {
                ret = min(ret, i - prev2);
            }
            prev1 = i;
        }
        else
        {
            if (prev1 != -1)
            {
                ret = min(ret, i - prev1);
            }
            prev2 = i;
        }
    }
    if(ret == -1)
    {
        cout << -1 << endl;
    }
    else
    {
        cout << ret << endl;
    }
    return 0;
}

输出结果：