Consider N coins aligned in a row. Each coin is showing either heads or tails. The adjacency of these coins is the number of adjacent pairs of coins with the same side facing up.
It must return the maximum possible adjacency that can be obtained by reversing exactly one coin (that is, one of the coins must be reversed). Consecutive elements of array A represent consecutive coins in the row. Array A contains only 0s and/or 1s: 0 represents a coin with heads facing up; 1 represents a coin with tails facing up. For example, given array A consisting of six numbers, such that:
A[0] = 1
A[1] = 1
A[2] = 0
A[3] = 1
A[4] = 0
A[5] = 0
the function should return 4. The initial adjacency is 2, as there are two pairs of adjacent coins with the same side facing up, namely (0, 1) and (4, 5). After reversing the coin represented by A[2], the adjacency equals 4, as there are four pairs of adjacent coins with the same side facing up, namely: (0, 1), (1, 2), (2, 3) and (4, 5), and it is not possible to obtain a higher adjacency. The same adjacency can be obtained by reversing the coin represented by A[3].
I should mention here that number of elements can be from 1....10'000. And it has to be done as O(N)
I attempted at the problem like so,
public static int adjacencies(int[] A) { int count = 0; boolean found = false; for(int i = 0; i < A.length-1; i++) { if (A[i] == A[i+1]) { System.out.println(A[i] + " " + A[i+1]); count++; } else if (((i + 2) < A.length-1) && !found) { if ((A[i] == 0 && A[i+1] == 1 && A[i+2] == 0) || (A[i] == 1 && A[i+1] == 0 && A[i+2] == 1)) { found = true; count = count + 2; } else if ((A[i] == 1 && A[i+1] == 0 && A[i+2] == 0) || (A[i] == 0 && A[i+1] == 1 && A[i+2] == 1)) { found = true; count = count + 1; } } } return count; } public static void main (String[] args) { int[] A = new int[] {1,1,0,1,0,0}; System.out.println(A); int count = adjacencies(A); System.out.println("Count: " + count); }
To me it looks right and it seems to work. Can someone find a better way of doing this instead of this clunky way ?