Amine-rich polymers for water purification applications