Update Compare_Version_Numbers.py

mcfair · Aug 11, 2015 · 8aaddbe · 8aaddbe
1 parent 11ebd99
commit 8aaddbe
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/Array/Compare_Version_Numbers.py b/Array/Compare_Version_Numbers.py
@@ -9,6 +9,13 @@
 
 0.1 < 1.1 < 1.2 < 13.37
 
+# https://leetcode.com/discuss/45331/2-4-lines-python-3-different-ways
+def compareVersion(self, version1, version2):
+    v1, v2 = (map(int, v.split('.')) for v in (version1, version2))
+    d = len(v2) - len(v1)
+    return cmp(v1 + [0]*d, v2 + [0]*-d)
+
+
 class Solution:
     # @param version1, a string
     # @param version2, a string