DSA/algorithms/Java/stacks/celebrity-problem.java

78 lines
2.0 KiB
Java
Raw Normal View History

//Problem Statement
// 1. You are given a number n, representing the number of people in a party.
// 2. You are given n strings of n length containing 0's and 1's
// 3. If there is a '1' in ith row, jth spot, then person i knows about person j.
// 4. A celebrity is defined as somebody who knows no other person than himself but everybody else knows him.
// 5. If there is a celebrity print it's index otherwise print "none".
//solution
import java.io.*;
import java.util.*;
public class Main {
public static void main(String[] args) throws Exception {
// write your code here
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
int n = Integer.parseInt(br.readLine());
int[][] arr = new int[n][n];
for (int j = 0; j < n; j++) {
String line = br.readLine();
for (int k = 0; k < n; k++) {
arr[j][k] = line.charAt(k) - '0';
}
}
findCelebrity(arr);
}
public static void findCelebrity(int[][] arr) {
// if a celebrity is there print it's index (not position), if there is not then print "none"
Stack < Integer > st = new Stack < > ();
for (int i = 0; i < arr.length; i++) {
st.push(i);
}
while (st.size() > 1) {
int i = st.pop();
int j = st.pop();
if (arr[i][j] == 1) {
st.push(j);
} else {
st.push(i);
}
}
int pot = st.pop();
boolean flag = true;
for (int i = 0; i < arr.length; i++) {
if (i != pot) {
if (arr[i][pot] == 0 || arr[pot][i] == 1) {
flag = false;
break;
}
}
}
if (flag) {
System.out.println(pot);
} else {
System.out.println("none");
}
}
}
//Test Case:-
//Input
// 4
// 0000
// 1011
// 1101
// 1110
//Output
//0