Codility编程课一binary gap

Task description

A binary gap within a positive integer N is any maximal sequence of consecutive zeros that is surrounded by ones at both ends in the binary representation of N.

For example, number 9 has binary representation 1001 and contains a binary gap of length 2. The number 529 has binary representation 1000010001 and contains two binary gaps: one of length 4 and one of length 3. The number 20 has binary representation 10100 and contains one binary gap of length 1. The number 15 has binary representation 1111 and has no binary gaps. The number 32 has binary representation 100000 and has no binary gaps.

Write a function:

def solution(N)

that, given a positive integer N, returns the length of its longest binary gap. The function should return 0 if N doesn’t contain a binary gap.

For example, given N = 1041 the function should return 5, because N has binary representation 10000010001 and so its longest binary gap is of length 5. Given N = 32 the function should return 0, because N has binary representation ‘100000’ and thus no binary gaps.

Assume that:
N is an integer within the range [1..2,147,483,647].
Complexity:

expected worst-case time complexity is O(log(N));
expected worst-case space complexity is O(1).

def bg(num):
  bif = bin(num)[2:]
  con_zero = 0
  max = 0
  for i in bif:
    if i == '0':
      con_zero += 1
    else:
      if con_zero>max:
        max = con_zero
      con_zero = 0
  return max

>>> bg(9)
2
>>> bg(1041)
5
>>> bg(32)
0
>>> bg(10)
1
>>> bg(529)
4
>>> bg(20)
1
>>> bg(15)
0