Update unique-word-abbreviation.cpp

kamyu104 · web-flow · commit c9c72d176425 · 2017-03-21T00:12:29.000+08:00
diff --git a/C++/unique-word-abbreviation.cpp b/C++/unique-word-abbreviation.cpp
@@ -21,6 +21,9 @@ class ValidWordAbbr {
     unordered_map<string, unordered_set<string>> lookup_;
 
     string abbreviation(const string& word) {
+        if (word.length() <= 2) {
+            return word;
+        }
         return word.front() + to_string(word.length()) + word.back();
     }
 };

Original file line number	Diff line number	Diff line change
`@@ -21,6 +21,9 @@ class ValidWordAbbr {`
`21`	`21`	`unordered_map<string, unordered_set<string>> lookup_;`
`22`	`22`
`23`	`23`	`string abbreviation(const string& word) {`
	`24`	`+ if (word.length() <= 2) {`
	`25`	`+ return word;`
	`26`	`+ }`
`24`	`27`	`return word.front() + to_string(word.length()) + word.back();`
`25`	`28`	`}`
`26`	`29`	`};`