请问字节现在算法题需要写main函数么