给一个函数，返回 0 和 1，概率为 p 和 1-p，请你实现一个函数，使得返回 01 概率一样。

解答思路：

根据题目的描述，我们需要实现一个函数，该函数以概率p返回0，以概率1-p返回1。然而，为了使得返回0和返回1的概率相同，我们可以利用某种方法来调整原始的p概率分布。一种简单的方法是取p的平均值来决定输出是0还是1。如果p等于或等于0.5，那么我们可以简单地随机返回0或1，因为此时两种结果出现的概率已经相等。如果p不等于0.5，我们可以先计算一个偏移量来调整原始的概率分布。如果p大于0.5，则更倾向于返回结果1，所以我们需要增加返回结果0的概率以平衡分布；反之亦然。因此，我们可以使用随机数生成器来根据这个偏移量决定函数的返回值。通过这种方式，我们可以实现一个函数使得返回0和1的概率大致相同。但是需要注意的是，这种方法的精确性取决于随机数生成器的质量以及p值的选择。对于极端的p值（如接近0或接近1），这种方法的准确性可能会降低。此外，这种方法并不能保证在每次调用函数时都能得到完全相等的概率分布，而是尽量接近相等的概率分布。

最优回答：

假设原始返回0和1的概率为p和1-p的函数为f()，我们的新函数g()可以如下实现：首先判断p与0.5的大小关系，如果p大于等于0.5，则直接随机返回0或1；否则计算偏移量并基于这个偏移量生成新的概率分布来决定返回值。具体实现时可以利用Python的random模块来生成随机数并判断概率分布。需要注意的是，由于随机性的影响，这种方法并不能保证每次调用函数时都能得到完全相等的概率分布。