可以用partition函数思想来做