Joe Root says England need to change their mental approach to batting for long periods, following their heavy first Test defeat to New Zealand.