额，3.5 ，4 都试过～发现 4 ，不仅能给出更好的代码，还能告诉我题目所隐藏的条件。。。顺便纠正我的错误描述。并最终给出，若你坚持这么做的话，可以加个 if 来判断的最终代码。
所以我的结论是，若不想专精 prompt ，用 4 会更好。