Task: Text to Code
Input (English): display the country ID and number of cities for each country.
Output (English): SELECT country_id ,  COUNT(*) FROM locations GROUP BY country_id