【LeetCode】 821. 字符的最短距离

Vector向量的应用

1. 题目介绍

题目
给定一个字符串 S 和一个字符 C。返回一个代表字符串 S 中每个字符到字符串 S 中的字符 C 的最短距离的数组。
示例 1:
输入: S = “loveleetcode”, C = ‘e’
输出: [3, 2, 1, 0, 1, 0, 0, 1, 2, 2, 1, 0]
说明:
1.字符串 S 的长度范围为 [1, 10000]。
2.C 是一个单字符，且保证是字符串 S 里的字符。
3.S 和 C 中的所有字母均为小写字母。

2. 思路分析
利用vector向量将距离存入向量中，再将向量返回

3. 代码

//C++
#include <iostream>
#include <vector>
#include <cmath>
using namespace std; 
vector<int> shortestToChar(string S, char C) {
	int len=S.length();
	vector<int>dis(len);
    for(int i=0;i<len;i++){
    	for(int j=i;j<len;j++){
    		if(S[j]==C){
    			dis[i]=abs(j-i);
    			break;
			}
			if(j==len-1){     //当后面没有的情况
				dis[i]=len;
			}
		}
		for(int j=i;j>=0;j--){
			if(S[j]==C){
    			dis[i]=min(abs(j-i),dis[i]);
    			break;
			}
		}
	}
	vector<int>::iterator t ;//向量的遍历输出 
    for(t=dis.begin(); t!=dis.end()-1; t++)
        cout<<*t<<",";
    cout<<*t<<endl;
	return dis;
}

int main(){
	string S="loveleetcode";
	char C='e';
	shortestToChar(S,C);
	return 0;
}

参考运行8ms的算法

class Solution
{
public:
    vector<int> shortestToChar(string S, char C)
    {
        int N = S.length();
        vector<int> vDistances(N, N);
        for ( int d = N, i = 0; i < N; ++i )
        {
            d = (C == S[i] ? 0 : d+1);
            //vDistances[i] = min(vDistances[i], d);
            vDistances[i] = d;
        }
        for ( int d = N, i = N-1; i >= 0; --i )
        {
            d = (C == S[i] ? 0 : d+1);
            vDistances[i] = min(vDistances[i], d);
        }
        return std::move(vDistances);
    }
};