Provable Burer-Monteiro factorization for a class of norm-constrained matrix problems