程序员代码面试指南刷题--第五章.数组中两个字符串的最小距离
程序员文章站
2024-02-27 23:47:09
...
题目描述
给定一个字符串数组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;
}
}