Python 高效解决 U 王国社区端午节粽子分配数量计算问题

这是一个典型的条件判断和累加计算问题。我们可以根据每户家庭人数确定分配的粽子数量，然后将所有户的粽子数量相加得到总数。 以下是Python的实现代码： ```python n = int(input()) total = 0 for _ in range(n): k = int(input()) if 1 <= k <= 3: total += 2 elif 4 <= k <= 6: total += 3 else: total += 5 print(total) ``` 代码解释： 1. 首先读取居民户数n 2. 初始化粽子总数total为0 3. 循环n次，每次读取一户的家庭人数k 4. 根据k的值判断应该分配的粽子数量： - 1-3人：加2个 - 4-6人：加3个 - 超过6人：加5个 5. 最后输出粽子总数total 这个程序的时间复杂度是O(n)，空间复杂度是O(1)，能够高效解决问题。 ######[AI问答 | 218点数解答 | 2025-05-31 18:33:52]