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(['0' 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