821. 字符的最短距离

Posted on | In
Words count in article: 418 | Reading time ≈ 2

题目

给定一个字符串 S 和一个字符 C。返回一个代表字符串 S 中每个字符到字符串 S 中的字符 C 的最短距离的数组.

示例1：

1
2
3
4
5
6
7
输入：s = "loveleetcode", c = "e"
输出：[3,2,1,0,1,0,0,1,2,2,1,0]
解释：字符 'e' 出现在下标 3、5、6 和 11 处（下标从 0 开始计数）。
距下标 0 最近的 'e' 出现在下标 3 ，所以距离为 abs(0 - 3) = 3 。
距下标 1 最近的 'e' 出现在下标 3 ，所以距离为 abs(1 - 3) = 2 。
对于下标 4 ，出现在下标 3 和下标 5 处的 'e' 都离它最近，但距离是一样的 abs(4 - 3) == abs(4 - 5) = 1 。
距下标 8 最近的 'e' 出现在下标 6 ，所以距离为 abs(8 - 6) = 2 。

示例2：

1
2
输入：s = "aaab", c = "b"
输出：[3,2,1,0]

提示：

  • 1 <= s.length <= 104
  • s[i] 和 c 均为小写英文字母
  • 题目数据保证 c 在 s 中至少出现一次

解法

解法一：

求出所有字符C在S中的索引，然后找到最小的值即可。

Java

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
class Solution {
    public int[] shortestToChar(String S, char C) {
		int[] result = new int[S.length()];
		List<Integer> indexs = new ArrayList<>();
        for (int i = 0;i < S.length();i++) {
        	if (C == S.charAt(i)) {
        		indexs.add(i);
        	}
        }
        
        for (int i = 0;i < S.length();i++) {
        	if (C == S.charAt(i)) {
        		result[i] = 0;
        	} else {
        		result[i] = getMin(i, indexs);
        	}
        }
        return result;
    }
	
	private int getMin(int index, List<Integer> indexs) {
		int min = Integer.MAX_VALUE;
		for (int i : indexs) {
			int value = Math.abs(i - index);
			if (min > value) {
				min = value;
			}
		}
		return min;
	}
}
0%