Python实现递归函数，返回多个值

题目描述：超市a和超市b，每次去超市a的人有20%下次仍然去超市a，其余80%去超市b；每次去超市b的人有40%仍去超市b，其余60%去超市a。若已知第一次去两个超市的人数，求第n次去两个超市的人数
输入：
第一行：去两个超市的次数
第二行：两个正整数 依次为 第一次去超市a和b的人数
输出：
一行 第n次去超市a和b的人数，向下取整

def get_nums(a,b,i,n):
    if i == n:
        return a,b
    else:
        return get_nums(int(0.2*a+0.6*b), int(0.8*b+0.4*b),i+1,n)

n = int(input())
a,b = map(int,input().split())
resa,resb=get_nums(a,b,1,n)
print(resa,resb)

import math


def computation(to_market_list):
    aa = to_market_list[0]
    bb = to_market_list[1]
    sum_market = to_market_list[2]

    if sum_market > 1:
        to_market_list[0] = math.floor(aa * 0.2) + math.floor(bb * 0.6)
        to_market_list[1] = math.floor(aa * 0.8) + math.floor(bb * 0.4)
        sum_market -= 1
        to_market_list[2] = sum_market
        computation(to_market_list)

    return to_market_list


def begin():
    sum_market = int(input())
    to_market_list = list(map(int, input().strip().split()))
    to_market_list.append(sum_market)

    res = computation(to_market_list)

    print(sum_market, res[0], res[1])


if __name__ == '__main__':
    begin()