Saturday, March 12, 2022

Perfect Friends

I/P:

7

5

0 1

2 3

4 5

5 6

4 6

O/P: 16




Step 1. Find connected components of graph

Step 2.  Using for loop, find pairing combinations


package pep.Day39;

import java.io.*;
import java.util.*;

public class Perfect_Friends {
static class Edge {
int src;
int nbr;

Edge(int src, int nbr) {
this.src = src;
this.nbr = nbr;
}
}

public static void main(String[] args) throws Exception {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

int vtces = Integer.parseInt(br.readLine());
ArrayList<Edge>[] graph = new ArrayList[vtces];
for (int i = 0; i < vtces; i++) {
graph[i] = new ArrayList<>();
}

int edges = Integer.parseInt(br.readLine());
for (int i = 0; i < edges; i++) {
String[] parts = br.readLine().split(" ");
int v1 = Integer.parseInt(parts[0]);
int v2 = Integer.parseInt(parts[1]);
graph[v1].add(new Edge(v1, v2));
graph[v2].add(new Edge(v2, v1));
}

boolean[] visited = new boolean[vtces];
ArrayList<ArrayList<Integer>> comps = new ArrayList<>();
ArrayList<Integer> comp;
for (int i = 0; i < vtces; i++) {
if (!visited[i]) {
comp = new ArrayList<>();
dfs(graph, i, visited, comp);
comps.add(comp);
}
}

int size = 0;
for (int i = 0; i < comps.size(); i++) {
for (int j = i+1; j < comps.size(); j++) {
size += (comps.get(i).size() * comps.get(j).size());
}
}

System.out.println(size);
}

private static void dfs(ArrayList<Edge>[] graph, int src, boolean[] visited, ArrayList<Integer> comp) {
visited[src] = true;
comp.add(src);
for (Edge e : graph[src]) {
if (!visited[e.nbr]) {
dfs(graph, e.nbr, visited, comp);
}
}
}

}






No comments:

Post a Comment

Diagonal Traversal

 eg.  1       2       3       4 5      6       7       8 9    10    11     12 13  14   15    16 Output: 1 6 11 16 2 7 12 3 8 4  Approach:...