代码写的漂亮。我跟你思想一样，用的set，但是代码写的累赘了。