-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsol.py
60 lines (43 loc) · 1.19 KB
/
sol.py
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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
from pwn import *
from itertools import combinations
context.log_level='debug'
p = remote("vending-machine.hsctf.com", 1337)
p.recvuntil(b"==\n")
p.sendline("Display")
p.recvuntil(b"Items: \n")
items = p.recvuntil(b"\n\n").decode()
items = items.split(":")
items = list(map(lambda x: int(x[:-2]), items[1:]))
print(items)
p.recvuntil(b"Coins: \n")
coins = p.recvuntil(b"\n\n").decode()
coins = coins.split(":")
coins = list(map(lambda x: int(x[:-2]), coins[1:]))
print(coins)
item_backup = items.copy()
coin_backup = coins.copy()
items.sort()
items.reverse()
coins = set(coins)
ans = []
for item in items[:-1]:
temp = 0
d = 99999999999999999999
for i in range(1, len(coins)):
print(i)
for li in combinations(coins, i):
s = sum(li)
if s > item and d > s - item:
temp = list(li)
d = s - item
coins -= set(temp)
print(coins)
ans.append(temp)
assert sum(coins) > items[-1] # try again
ans.append(list(coins))
print(ans)
for i, li in enumerate(ans):
for c in li:
p.sendline(f"Insert {coin_backup.index(c) + 1}")
p.sendline(f"Buy {item_backup.index(items[i]) + 1}")
p.interactive()