LintCode: Longest Words

C++

class Solution {
public:
    /**
     * @param dictionary: a vector of strings
     * @return: a vector of strings
     */
    vector<string> longestWords(vector<string> &dictionary) {
        // write your code here
        if (dictionary.size() <= 1) {
            return dictionary;
        }
        
        vector<string> result;
        int max_len = 0, cur_len;
        
        for (int i=0; i<dictionary.size(); i++) {
            cur_len = dictionary[i].size();
            if(cur_len < max_len) {
                continue;
            }
            if(cur_len > max_len) {
                result.clear();
                max_len = cur_len;
            }
            result.push_back(dictionary[i]);
        }
        return result;
    }
};