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() <= ) {
             return dictionary;
         }

         vector<string> result;
         int max_len = , cur_len;

         for (int i=; 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;
     }
 };