Water resources management is closely related to environmental protection and other disciplines, a large amount of data and various indicators. With the continuous development of Water Pollution (WP) control technology, WP prevention and control is diverse and systematic. However, technical improvement is still needed in terms of expressiveness, integration and authority. In terms of the sustainable development of water resources, it is necessary to make short-term and medium-long term predictions of water resources, and make appropriate improvements to WP control planning technologies such as the optimization of planning schemes. The comprehensive utilization, development and utilization of water resources in river basins can lay a solid foundation for the formulation of national water resources protection policies and implementation plans, and the coordination of national, regional, river basin and department water resources management. Chaotic particle swarm optimization has the advantages of fast convergence, less control, simple implementation, and avoiding local bias. This method can effectively reduce the order and make the model closer to the original system. Through the analysis of experimental data, it was found that the operation efficiency of chaotic particle swarm optimization algorithm was 36.84% higher than that of traditional particle swarm optimization algorithm, and the operation accuracy of chaotic particle swarm optimization algorithm was 10.4% higher than that of traditional particle swarm optimization algorithm.