Compare Versions

compare_versions.rb

class MySolution
    # @param a : string
    # @param b : string
    # @returns an integer
    def compare_versions(a, b)
        aa = a.split('.')
        bb = b.split('.')
        
        i = 0

        loop do
            if i < aa.length && i < bb.length
                if aa[i].to_i > bb[i].to_i
                    return 1
                elsif aa[i].to_i < bb[i].to_i
                    return -1
                end
            else
                if i < aa.length && aa[i].to_i.nonzero?
                    return 1
                elsif i < bb.length && bb[i].to_i.nonzero?
                    return -1
                else
                    return 0
                end
            end
            
            i += 1
        end
    end
end