USA:s president Donald Trump säger att han väntar sig att landets militär kommer att lämna Iran om ”två till tre veckor”.