Through the Analysis and comparison of the traditional Cluster - Tree algorithm, AODVjr algorithm and Cluster-Tree+AODVjr algorithm of the ZigBee protocol, a method of“handshake”optimization algorithm is presented for the special system of freight locomotive. This algorithm realizes the data transmission through the handshake connection between two nodes, and it not only saves the energy consumption of nodes, but also overcomes the disadvantages of ZigBee technology in short distance transmission.