Pythonで敵が同じグループにとどまることができない人を分離するプログラム

数nと敵と呼ばれる2D行列があるとします。ここで、nは、[0、n-1]からラベル付けされたn人がいることを示します。これで、敵の各行に[a、b]が含まれます。これは、aとbが敵であることを意味します。敵である2人が同じグループに入らないように、n人を2つのグループに分割できるかどうかを確認する必要があります。

したがって、入力がn =4、敵=[[0、3]、[3、2]]の場合、これら2つのグループ[0、1、2]と[を使用できるため、出力はTrueになります。 3]。

これを解決するには、次の手順に従います-

• グラフ：=空の隣接リスト

• 敵の各敵ペア（u、v）について、実行します

  • グラフの最後にvを挿入します[u]

  • グラフの最後にuを挿入します[v]

• 色：=新しい地図

• 関数dfs（）を定義します。これにはu、c：=0が最初にかかります

• uがカラーの場合、

  • color[u]がc

    と同じ場合はtrueを返します

• color [u]：=c

• グラフ[u]の各vのすべてのdfs（v、c XOR 1）がtrueの場合、trueを返します

• メインの方法から、次のようにします-

• （0からnの範囲の各uのdfs（u）で、uがカラーでない場合）がすべてtrueの場合にtrueを返します

理解を深めるために、次の実装を見てみましょう-

例

class Solution:
   def solve(self, n, enemies):
      from collections import defaultdict
      graph = defaultdict(list)
      for u, v in enemies:
         graph[u].append(v)
         graph[v].append(u)
      color = {}
      def dfs(u, c=0):
         if u in color:
            return color[u] == c
            color[u] = c
            return all(dfs(v, c ^ 1) for v in graph[u])
         return all(dfs(u) for u in range(n) if u not in color)
ob = Solution()
n = 4
enemies = [[0, 3],[3, 2]]
print(ob.solve(n, enemies))

入力

4, [[0, 3],[3, 2]]

出力

True