改了一下算法

Signed-off-by: boddmg <boddmg@gmail.com>