在Leetcode等刷题网站刷题时,为了方便少敲分号,我经常会选择用python;但头疼的是,同样的算法有时候C++能过但python不能过,比如关于Python答题时报maximum recursion depth exceeded in comparison
,往往是因为用了递归算法,当深度过深时被python虚拟机直接拒绝了。
举例
牛客题目地址:https://www.nowcoder.com/practice/52f25c8a8d414f8f8fe46d4e62ef732c
题目描述
小多想在美化一下自己的庄园。他的庄园毗邻一条小河,他希望在河边种一排树,共 M M M 棵。小多采购了 N N N 个品种的树,每个品种的数量是 A i A_i Ai (树的总数量恰好为 M M M)。但是他希望任意两棵相邻的树不是同一品种的。小多请你帮忙设计一种满足要求的种树方案。
输入描述:
第一行包含一个正整数