競プロおぼえがき

おぼえがきのためのブログ

yukicoder | No.488 四角関係

  • 問題

No.488 四角関係 - yukicoder

  • 概要

 N個のノードと各ノード関係がM個渡される。
ノードの中で四角の関係(4点の閉路グラフになっているもの。3点で作れる場合は不可)になっているものの数を出力する。

  • 考え方

全探索をする。4つのノードa, b, c, dを選んで、選んだノードすべてが選ばれたノードの中で2つのノードだけと関係を持っている場合は四角の関係になっていると判定する。

import java.util.Scanner;

class Main {
    static Scanner in = new Scanner(System.in);
    static boolean[][] r;

    public static void main(String[] args) {
        int N = in.nextInt(), M = in.nextInt();
        r = new boolean[N][N];
        for (int i = 0; i < M; i++) {
            int a = in.nextInt(), b = in.nextInt();
            r[a][b] = r[b][a] = true;
        }
        int sum = 0;
        for (int a = 0; a < N - 3; a++) {
            for (int b = a + 1; b < N - 2; b++) {
                for (int c = b + 1; c < N - 1; c++) {
                    for (int d = c + 1; d < N; d++) {
                        sum += rect(new int[] { a, b, c, d });
                    }
                }
            }
        }
        System.out.println(sum);
    }

    public static int rect(int[] n) {
        boolean b = true;
        for(int i = 0; i < 4; i++){
            int a = 0;
            for(int j = 0; j < 4; j++){
                if(i == j) continue;
                if(r[n[i]][n[j]]) a++;
            }
            if(a != 2) b = false;
        }
        return b ? 1 : 0;
    }
}