版权声明:made by YYT https://blog.csdn.net/qq_37621506/article/details/83064205
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);
}
};