Ugo La Malfa, la ricostruzione delle ferrovie e la riforma dello Stato