p2_new.py

"""f = open('b1.in', 'r')
lines = f.readlines()
index = 0
num_tests = int(lines[index].strip())
index += 1 #move to blank line"""

last_call_ans = 0
ir = 43

def move_left(nums):
	new_nums = [0]*size_strip
	inPair = False
	pastNum = 0
	index = 0
	for i, num in enumerate(nums):
		if num == 0:
			continue
		else:
			if inPair and pastNum == num:
				new_nums[index] = pastNum*2
				index += 1
			elif inPair and pastNum != num:
				new_nums[index] = pastNum
				new_nums[index+1] = num
				inPair = False
				index += 2
			elif not inPair:
				pastNum = num
				inPair = True
	return new_nums

def move_right(nums):
	new_nums = [0]*size_strip
	inPair = False
	pastNum = 0
	index = len(new_nums)-1
	nums.reverse()
	pairs = enumerate(nums)
	for i, num in pairs:
		if num == 0:
			continue
		else:
			if inPair and pastNum == num:
				new_nums[index] = pastNum*2
				index -= 1
			elif inPair and pastNum != num:
				new_nums[index] = pastNum
				new_nums[index-1] = num
				inPair = False
				index -= 2
			elif not inPair:
				pastNum = num
				inPair = True
	return new_nums

def random_shit():
	global ms
	global ir
	global last_call_ans	
	c = 1 if ((ir>=43) and last_call_ans < 0) else 0
	ans = (ms[ir-22] - ms[ir-43] - c) % (2**32)
	last_call_ans = ans
	ir += 1
	return ans

with open('b1.in','r') as infile:
	num_test_cases = int(infile.next().strip())
	print num_test_cases
	for i in range(0, num_test_cases):
		infile.next()
		size_strip = int(infile.next().strip())
		print "size_strip", size_strip
		nums = infile.next().strip().split()
		print "nums", nums
		ms = infile.next().strip().split()
		ms = [int(m) for m in ms]
		num_moves = int(infile.next().strip())
		print "num_moves", num_moves
		moves = list(infile.next().strip())
		print "moves", moves

		for move in moves:
			if move == 'l':
				new_nums = move_left(nums)
			else:
				new_nums = move_right(nums)

			#get new random val
			num_empty = sum([1 for x in nums if x == 0])
			if new_nums == nums:
				if num_empty==0:
					new_nums = new_nums
				else:
					pos = random_shit() % num_empty
					if (random_shit() % 10) == 0:
						new_value = 4
					else:
						new_value = 2

			#insert it
				empty_counter = 0
				for i,num in enumerate(new_nums):
					if new_nums[i] != 0:
						continue
					if new_nums[i] == 0 and empty_counter == pos:
						new_nums[i] = new_value
					else:
						empty_counter += 1

			nums = new_nums



	"""while index < len(lines):
	index += 1 #move past blank line
	size_strip = int(lines[index].strip())
	print size_strip
	index += 1 #move past size strip line
	nums = lines[index].strip().split()
	nums = [int(num) for num in nums] #the numbers in the 2048 grid (row)
	print nums
	index += 1 #move past strip nums
	ms = lines[index].strip().split()	
	ms = [int(m) for m in ms]
	index += 1
	num_moves = int(lines[index])
	index += 1
	moves = lines[index]"""