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