给定一个字符串 S 和一个字符 C。返回一个代表字符串 S 中每个字符到字符串 S 中的字符 C 的最短距离的数组。
eg:
输入: S = “loveleetcode”, C = ‘e’
输出: [3, 2, 1, 0, 1, 0, 0, 1, 2, 2, 1, 0]
方法一:
是最简单也是最笨的,即用两个嵌套for循环,遍历其中每个元素,并选取当前元素到左边与右边和他相等元素的最小距离。具体代码如下:
class Solution {
public int[] shortestToChar(String S, char C) {
int arr[]=new int[S.length()];
for(int i=0;i<arr.length;i++){
int num=S.length()+1;
int sum=S.length()+1;
for(int j=i;j<arr.length;j++){
if(S.charAt(j)==C){
num=j-i;
break;
}
}
for(int k=i;k>=0;k--){
if(S.charAt(k)==C){
sum=i-k;
break;
}
}
arr[i]=Math.min(num,sum);
}
return arr;
}
}
方法二:
对比方法一降低了时间复杂度,用两个并列for循环来完成。具体思路是先创建长度相等的数组,然后从字符串的第一个字符开始找,并判断是否和目标字符相同,若相同从这个角标开始对应数组的角标和角标里的数每次都加1,即 arr[a++]=b++;然后第二遍进行倒序循环从最后一个元素开始,当找和目标字符相等的字符时,当前角标对应的数组的角标随着角标数减小角标内对应数字增大,当大于原有数字时,则数组不进行重新赋值,当第二次循环到第一个与目标字符相等 的角标时,数组角标对应的值应为当前角标减去循环的次数。具体代码如下:
class Solution {
public int[] shortestToChar(String S, char C) {
int [] arr=new int[S.length()];
int a=0,b=0,c=0; //c表示第一个与字符C相等 的字符串对应的角标
boolean flag=false;
for(int i=0;i<S.length();i++){
if(S.charAt(i)==C){
b=0,a=i;
flag=true;
}
if(flag){
arr[a++]=b++;
}else{
c++;
}
}
for(int i=S.length()-1;i>=0;i--){
if(i<=c){
arr[i]=c-i;
continue;
}
if(S.charAt(i)==C){
flag=false;
b=0,a=i;
}
if(!flag){
arr[a]=Math.min(b++,arr[a--]);
}
}
return arr;
}
}