求一个字符串中,每个位置到指定字符的最短距离
我能想到的常规做法,即开辟一个专门的容器用于记录每个指定字符出现的位置。但需要对开始以及结尾的位置单独讨论,感觉这么做挺麻烦的
考虑到题目数据范围:字符串长度在1~10000之间,这就意味着,返回的容器中,每个值都不会超过10000
于是,在记录指定字符位置的容器的头尾分别插入-20000和30000的值,这样在遍历字符串时,只要单纯考虑当前位置距离前后指定字符出现位置的最小值就可以了
class Solution {
public:
vector<int> shortestToChar(string S, char C) {
int i,j=1;
vector<int> mark,ans(S.size(),0);
mark.push_back(-20000);
for(i=0;i<S.size();i++)
if(S[i]==C)
mark.push_back(i);
mark.push_back(30000);
for(i=0;i<S.size();i++)
{
if(i==mark[j])
{
ans[i]=0;
j++;
}
else
{
ans[i]=min(i-mark[j-1],mark[j]-i);
}
}
return ans;
}
};