flag
软件学院大三党,每天一道算法题,第28天
题目介绍
给定一个字符串 S 和一个字符 C。返回一个代表字符串 S 中每个字符到字符串 S 中的字符 C 的最短距离的数组。
思路
仔细观察输出数组,发现是一个有峰值的序列,我们首先要找到所有0的位置,然后从左到右遍历一次,找到每个值离左面最近0的距离,再从右到左遍历,找到每个值里右面最近0的距离,如果更小就覆盖之前的值
关键代码
public static int[] shortestToChar(String S, char C){
char []arr=S.toCharArray();
int len=arr.length;
int []result=new int[len];//保存结果
for (int i=0;i<len;i++){//初始化,将C对应的结果设为0,其它设为max_Value
if(arr[i]==C)
result[i]=0;
else
result[i]=Integer.MAX_VALUE;
}
int count=-1;//计数器,记录两个C直接的距离,不能初始化为0,因为arr[0]可能为C
for(int i=0;i<len;i++){//从左到右遍历一次
if(result[i]==0){
count=i;
}
else if(result[i]!=0&&count!=-1) {
result[i]=i-count;
}
}
count=0;//初始化
for(int i=len-1;i>=0;i--){//从右往左遍历
if(result[i]==0){
count=i;
}
else if(result[i]!=0&&count!=0&&result[i]>count-i)
result[i]=count-i;
}
return result;
}