3.21 PAT1078 乙级 python

字符串压缩与解压

x = input()
y = input()
def yasuo(a):
	k = a[0]
	lst = []
	count = 0
	for i in range(len(a)):
		if a[i] == k:
			count += 1
			if i ==len(a)-1:
				lst.append(str(count))
				lst.append(k)
		else:
			if count == 1:
				lst.append(k)
			else:
				lst.append(str(count))
				lst.append(k)
			k = a[i]
			count = 1
			if i ==len(a)-1:
				if count !=1:
					lst.append(str(count))
					lst.append(k)
				else:
					lst.append(k)
	return lst
def jieya(a):
	result = ''
	i = 0
	while i<len(a):
		j = i+1
		if a[i].isdigit():
			while a[j].isdigit():
				j+=1
			num = int(a[i:j])
			result+=num*a[j]
			i=j+1
		else:
			result += a[i]
			i+=1
	return list(result)
if x=='C':
	print(''.join(yasuo(y)))
else:
	print(''.join(jieya(y)))