优化了一下数据结构

现在可以方便地调用了。