Submissions for team Yibo's Wang on problem nah_main