Country and State are tightly connected even if you generate the default layout - you can't separate state from the country without custom code. You get $$input:AddressCountryID$$
, State is not rendered in the custom form layout.
Although in terms of HTML - Kentico rednders 2 div tags:
<div id="...AddressCountryID_uniSelectorCountry">
<div id="...AddressCountryID_uniSelectorState">
Since it is purely visual change I would just switch them using jQuery:
$("div[id*='addressForm_AddressCountryID_uniSelectorState']").remove().insertBefore("div[id*='addressForm_AddressCountryID_uniSelectorCountry']")