程序不难写，但花了几个小时优化。