學校作業的code，有興趣的自己拿去研究