程序员代码面试指南刷题--第五章.数组中两个字符串的最小距离

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

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

输出描述:

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

示例1

输入
复制

1
CD AB
CD

输出
复制

-1

示例2

输入

5
QWER 666
QWER
1234
qwe
666
QWER

输出

1

解法一：遍历数组

import java.io.*;
import java.util.*;
public class Main{
    public static void main(String[] args) throws Exception{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int len = Integer.parseInt(br.readLine());
        String[] info = br.readLine().trim().split(" ");
        String[] arr = new String[len];
        for(int i=0;i<len;i++){
            arr[i] = br.readLine().trim();
        }
        int res = getRes(arr,info[0],info[1]);
        System.out.println(res);
    }
    public static int getRes(String[] arr,String s1,String s2){
        if(s1==null||s2==null) return -1;
        int lasts1 = -1;
        int lasts2 = -1;
        int min = Integer.MAX_VALUE;
        for(int i=0;i<arr.length;i++){
            if(arr[i].equals(s1)){
                lasts1 = i;
                if(lasts2!=-1){
                    min = Math.min(min,lasts1-lasts2);
                }
            }
            if(arr[i].equals(s2)){
                lasts2 = i;
                if(lasts1!=-1){
                    min = Math.min(min,lasts2-lasts1);
                }
            }
        }
        return lasts1==-1||lasts2==-1?-1:min;
    }
}