请问算法题是需要自己导包，写main函数吗？还是像力扣那样写函数就可以了