해커랭크(HackerRank) - Weighted Uniform Strings / C++

 

문제 : 해커랭크 - Weighted Uniform Strings

난이도: Easy

언어 : C+

// Complete the weightedUniformStrings function below.
vector<string> weightedUniformStrings(string s, vector<int> queries) {
    
    vector<string> result;
    map<int,int> weight;
    
    char prev = 0;
    int prevWeight = 0;
    for(int i = 0 ;i<s.length();i++)
    {
        if(prev == s[i])
        {   
            prevWeight+=(s[i]-'a'+1);
        }
        else
        {
            prevWeight =(s[i]-'a'+1);
          
        }
        weight[prevWeight] = prevWeight; 
       
        prev = s[i]; 
    }

    for(int i = 0 ;i<queries.size();i++)
    {
       if( weight.find(queries[i]) != weight.end())
       {
           result.push_back("Yes");
       }else
            result.push_back("No");
    }
    return result;
}