about summary refs log tree commit diff
path: root/users/wpcarro/scratch/data_structures_and_algorithms/bit-manipulation.py
blob: dc30bb5088873e31f4bc57e2b5faf5ac9beebcaf (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
def test(x, i):
    return x & (1 << i) != 0


def set(x, i):
    return x | (1 << i)


def clear(x, i):
    return x & ~(1 << i)


def toggle(x, i):
    if test(x, i):
        return clear(x, i)
    else:
        return set(x, i)


def test_single(x):
    if x == 0:
        return False
    else:
        return x & (x - 1) == 0


print(test(0b1010, 3))
print('{0:b}'.format(set(0b1010, 1)))
print('{0:b}'.format(clear(0b1010, 1)))
print('{0:b}'.format(toggle(0b1010, 2)))
print(test_single(0b1010))
print(test_single(0b1000))