Government forces in Afghanistan have retaken the control of a remote district in the country’s northern province of Badakhshan from Taliban militants, officials say.