Problem:Given a string S
and a character C
, return an array of integers representing the shortest distance from the character C
in the string.
题目:给定一个字符串S和字符C,返回一个整型数组,数组每一位表示字符串对应下标位字符距离C字符的最短距离。
思路:首先,以key表示开头第一位或者后面最后一次遇到C字符的下标;当再一次遇到C字符时,处理返回数组的值;ret[]为返回数组,i 为遍历指针。
1,遍历S字符串,ret[i]赋值为-1,表示该位待处理。
2,当遇到C字符时,ret[i] 赋值该位为0;同时如果key下标位ret[key]!=0时,key为第一位,k=i;即左边无C字符,直接遍历数组到i位,依次赋值k--
3,若ret[key]=0;说明key位为中间某个位置,k=1;同时从两头遍历key到i之间的数组,分别赋值k++;直到两头指针相遇停止;
4,循环1-3;直到结束
5,判断最后一位为是否为0;若不为0则表示,最后一位不是C字符,则,key位到最后一位还未处理;k=1;ret[key++]=k++;返回结果ret
6,若最后一位为0;算法结束,返回ret
代码:
class Solution {
public int[] shortestToChar(String S, char C) {
int[] ret = new int[S.length()];
int key = 0;
int flag = 0;
int i = 0;
for (char c : S.toCharArray()) {
ret[i] = -1;
if (c == C) {
ret[i] = 0;
if (ret[key] != 0) {
int k = i;
while (key < i) {
ret[key] = k;
key++;
k--;
}
}
if (ret[key] == 0)
key++;
int x = 1;
flag = i;
for (int j = flag - 1, k = key; j >= k ; j--, k++) {
ret[j] = x;
ret[k] = x;
x++;
}
key=i;
}
i++;
}
if(ret[i-1]!=0)
{
flag =1;
key++;
for(;key<i;key++) {
ret[key]=flag;
flag++;
}
}
return ret;
}
}