The head of Wal-Mart Stores' business in China is leaving the company as it deals with a pricing probe that has led to stores being closed and staff arrested.