自定义k-e方程代码