春招的金三银四呀我们来做个有挑战性的题目吧

一、题目内容

背景：long time ago，大地上有一些城市和一些道路，城市之间能够互通的称为国家，不同的国家的城市之间是无法相通的。如图的例子所示就是两个国家：蓝国和红国：

问题：现在我们拿到了一批数据，请从数据中判断有多少个国家，并且哪些城市是同为一个国家的.
数据格式：数据的格式是以 [国家c，道路p] 这样的两个元素表示的，表示这个国家a是连在这道路p上的。
上图所示用数据就可以表示为
[c0，p2]
[c1，p1]
[c2，p1]
[c3，p1]
[c4，p3]

二、解题模板

python:

def devide_city(data):
     '''
    :param data: 例子：[['c0', 'p2'], ['c1', 'p1'], ['c2', 'p1']]
    :return: 返回list，list中的元素是每个国家的城市集合，例子：[['c0'], ['c1', 'c2']] 
     '''
    pass

sparksql:
表：city_path
两个string类型的字段city和path

这是之前的题目 如果你还想挑战下 https://www.jianshu.com/p/f9d2e7e58c01