Leetcode OJ : Compare Version Numbers Python solution

Total Accepted: 12400 Total Submissions: 83230

 

 

Compare two version numbers version1 and version2.
If version1 > version2 return 1, if version1 < version2 return -1, otherwise return 0.

You may assume that the version strings are non-empty and contain only digits and the . character.
The . character does not represent a decimal point and is used to separate number sequences.
For instance, 2.5 is not "two and a half" or "half way to
version three", it is the fifth second-level revision of the second
first-level revision.

Here is an example of version numbers ordering:

0.1 < 1.1 < 1.2 < 13.37

Credits:
Special thanks to @ts for adding this problem and creating all test cases.

Solution:

 class Solution:
     # @param version1, a string
     # @param version2, a string
     # @return an integer
     def compareVersion(self, version1, version2):
         splited1, splited2 = version1.split('.'), version2.split('.')
         diff = len(splited1) - len(splited2)

         ext = splited1 if diff < 0 else splited2;
         ext.extend(['' for i in range(abs(diff))])

         for a, b in zip(splited1, splited2):
             ret = cmp(int(a), int(b))
             if ret != 0:
                 return ret
         return 0