package template;

import java.util.ArrayList;
import java.util.Map;
import java.util.HashMap;

public class TestCase {

    private boolean isSolved;
    private Object solution;
    private Map<String, Integer> intProperties;
    private Map<String, ArrayList<Integer>> intArrayProperties;
    private Map<String, ArrayList<ArrayList<Integer>>> intArray2DProperties;
    private Map<String, Double> doubleProperties;
    private Map<String, ArrayList<Double>> doubleArrayProperties;
    private Map<String, ArrayList<ArrayList<Double>>> doubleArray2DProperties;
    private Map<String, String> stringProperties;
    private Map<String, ArrayList<String>> stringArrayProperties;
    private Map<String, ArrayList<ArrayList<String>>> stringArray2DProperties;
    private Map<String, Boolean> booleanProperties;
    private Map<String, ArrayList<Boolean>> booleanArrayProperties;
    private Map<String, ArrayList<ArrayList<Boolean>>> booleanArray2DProperties;
    private Map<String, Long> longProperties;
    private Map<String, ArrayList<Long>> longArrayProperties;
    private Map<String, ArrayList<ArrayList<Long>>> longArray2DProperties;
    private int ref;
    private double time;

    public TestCase() {
        initialise();
    }

    private void initialise() {
        isSolved = false;
        intProperties = new HashMap<>();
        intArrayProperties = new HashMap<>();
        intArray2DProperties = new HashMap<>();
        doubleProperties = new HashMap<>();
        doubleArrayProperties = new HashMap<>();
        doubleArray2DProperties = new HashMap<>();
        stringProperties = new HashMap<>();
        stringArrayProperties = new HashMap<>();
        stringArray2DProperties = new HashMap<>();
        booleanProperties = new HashMap<>();
        booleanArrayProperties = new HashMap<>();
        booleanArray2DProperties = new HashMap<>();
        longProperties = new HashMap<>();
        longArrayProperties = new HashMap<>();
        longArray2DProperties = new HashMap<>();
        ref = 0;
    }

    public void setSolution(Object o) {
        solution = o;
        isSolved = true;
    }
    
    public Object getSolution() {
        if (!isSolved) {
            Utils.die("getSolution on unsolved testcase");
        }

        return solution;
    }

    public void setRef(int i) {
        ref = i;
    }
    public int getRef() {
        return ref;
    }

    public void setTime(double d) {
        time = d;
    }
    public double getTime() {
        return time;
    }
    
    public void setInteger(String s, Integer i) {
        intProperties.put(s, i);
    }
    
    public Integer getInteger(String s) {
        return intProperties.get(s);
    }

    public void setIntegerList(String s, ArrayList<Integer> l) {
        intArrayProperties.put(s, l);
    }

    public void setIntegerMatrix(String s, ArrayList<ArrayList<Integer>> l) {
        intArray2DProperties.put(s, l);
    }

    public ArrayList<Integer> getIntegerList(String s) {
        return intArrayProperties.get(s);
    }

    public Integer getIntegerListItem(String s, int i) {
        return intArrayProperties.get(s).get(i);
    }

    public ArrayList<ArrayList<Integer>> getIntegerMatrix(String s) {
        return intArray2DProperties.get(s);
    }
    
    public ArrayList<Integer> getIntegerMatrixRow(String s, int row) {
        return intArray2DProperties.get(s).get(row);
    }
    
    public Integer getIntegerMatrixItem(String s, int row, int column) {
        return intArray2DProperties.get(s).get(row).get(column);
    }

    public ArrayList<Integer> getIntegerMatrixColumn(String s, int column) {
        ArrayList<Integer> out = new ArrayList();
        
        for(ArrayList<Integer> row : intArray2DProperties.get(s)) {
            out.add(row.get(column));
        }
        return out;
    }


    public void setDouble(String s, Double i) {
        doubleProperties.put(s, i);
    }
    
    public Double getDouble(String s) {
        return doubleProperties.get(s);
    }

    public void setDoubleList(String s, ArrayList<Double> l) {
        doubleArrayProperties.put(s, l);
    }

    public void setDoubleMatrix(String s, ArrayList<ArrayList<Double>> l) {
        doubleArray2DProperties.put(s, l);
    }

    public ArrayList<Double> getDoubleList(String s) {
        return doubleArrayProperties.get(s);
    }

    public Double getDoubleListItem(String s, int i) {
        return doubleArrayProperties.get(s).get(i);
    }

    public ArrayList<ArrayList<Double>> getDoubleMatrix(String s) {
        return doubleArray2DProperties.get(s);
    }
    
    public ArrayList<Double> getDoubleMatrixRow(String s, int row) {
        return doubleArray2DProperties.get(s).get(row);
    }
    
    public Double getDoubleMatrixItem(String s, int row, int column) {
        return doubleArray2DProperties.get(s).get(row).get(column);
    }

    public ArrayList<Double> getDoubleMatrixColumn(String s, int column) {
        ArrayList<Double> out = new ArrayList();
        
        for(ArrayList<Double> row : doubleArray2DProperties.get(s)) {
            out.add(row.get(column));
        }
        return out;
    }

    
    public void setString(String s, String t) {
        stringProperties.put(s, t);
    }
    
    public String getString(String s) {
        return stringProperties.get(s);
    }

    public void setStringList(String s, ArrayList<String> l) {
        stringArrayProperties.put(s, l);
    }

    public void setStringMatrix(String s, ArrayList<ArrayList<String>> l) {
        stringArray2DProperties.put(s, l);
    }

    public ArrayList<String> getStringList(String s) {
        return stringArrayProperties.get(s);
    }

    public String getStringListItem(String s, int i) {
        return stringArrayProperties.get(s).get(i);
    }

    public ArrayList<ArrayList<String>> getStringMatrix(String s) {
        return stringArray2DProperties.get(s);
    }
    
    public ArrayList<String> getStringMatrixRow(String s, int row) {
        return stringArray2DProperties.get(s).get(row);
    }
    
    public String getStringMatrixItem(String s, int row, int column) {
        return stringArray2DProperties.get(s).get(row).get(column);
    }

    public ArrayList<String> getStringMatrixColumn(String s, int column) {
        ArrayList<String> out = new ArrayList();
        
        for(ArrayList<String> row : stringArray2DProperties.get(s)) {
            out.add(row.get(column));
        }
        return out;
    }

    
    public void setBoolean(String s, Boolean b) {
        booleanProperties.put(s, b);
    }
    
    public Boolean getBoolean(String s) {
        return booleanProperties.get(s);
    }

    public void setBooleanList(String s, ArrayList<Boolean> l) {
        booleanArrayProperties.put(s, l);
    }

    public void setBooleanMatrix(String s, ArrayList<ArrayList<Boolean>> l) {
        booleanArray2DProperties.put(s, l);
    }

    public ArrayList<Boolean> getBooleanList(String s) {
        return booleanArrayProperties.get(s);
    }

    public Boolean getBooleanListItem(String s, int i) {
        return booleanArrayProperties.get(s).get(i);
    }

    public ArrayList<ArrayList<Boolean>> getBooleanMatrix(String s) {
        return booleanArray2DProperties.get(s);
    }
    
    public ArrayList<Boolean> getBooleanMatrixRow(String s, int row) {
        return booleanArray2DProperties.get(s).get(row);
    }
    
    public Boolean getBooleanMatrixItem(String s, int row, int column) {
        return booleanArray2DProperties.get(s).get(row).get(column);
    }

    public ArrayList<Boolean> getBooleanMatrixColumn(String s, int column) {
        ArrayList<Boolean> out = new ArrayList();
        
        for(ArrayList<Boolean> row : booleanArray2DProperties.get(s)) {
            out.add(row.get(column));
        }
        return out;
    }

    public void setLong(String s, Long b) {
        longProperties.put(s, b);
    }
    
    public Long getLong(String s) {
        return longProperties.get(s);
    }

    public void setLongList(String s, ArrayList<Long> l) {
        longArrayProperties.put(s, l);
    }

    public void setLongMatrix(String s, ArrayList<ArrayList<Long>> l) {
        longArray2DProperties.put(s, l);
    }

    public ArrayList<Long> getLongList(String s) {
        return longArrayProperties.get(s);
    }

    public Long getLongListItem(String s, int i) {
        return longArrayProperties.get(s).get(i);
    }

    public ArrayList<ArrayList<Long>> getLongMatrix(String s) {
        return longArray2DProperties.get(s);
    }
    
    public ArrayList<Long> getLongMatrixRow(String s, int row) {
        return longArray2DProperties.get(s).get(row);
    }
    
    public Long getLongMatrixItem(String s, int row, int column) {
        return longArray2DProperties.get(s).get(row).get(column);
    }

    public ArrayList<Long> getLongMatrixColumn(String s, int column) {
        ArrayList<Long> out = new ArrayList();
        
        for(ArrayList<Long> row : longArray2DProperties.get(s)) {
            out.add(row.get(column));
        }
        return out;
    }

    

    


}

package com.vp.impl;

import java.util.Arrays;

import com.vp.common.CommonUtility;
import com.vp.iface.Problem;

public class DancingWithGooglers implements Problem {

	private static int currentNoOfSurprises;
	private static int eligibleGooglers;
	private static final int noOfJudges = 3;
	private static int noOfGooglers;
	private static int noOfSurprises;
	private static int bestScore;
	
	@Override
	public String solve(String[] dataset) {
		String strReturnValue ="";
		String strDanceData = dataset[0];
		String strDanceDataArr[] = strDanceData.split(" ");
		 int iDanceDataArr[] = CommonUtility.convertStringArraytoIntArray(strDanceDataArr);
		 
		  noOfGooglers = iDanceDataArr[0];
		  noOfSurprises = iDanceDataArr[1];
		  bestScore = iDanceDataArr[2];
		  eligibleGooglers = 0;
		  currentNoOfSurprises =0;
		  
		 int totalScore;
		 int googlers[] = new int [noOfGooglers];
		 int index =0 ;
		 int scoreSplit[] = new int[noOfJudges];
		 
		 CommonUtility.print(iDanceDataArr);
		 for (int i = 3; i < iDanceDataArr.length; i++ )
		 {
			 googlers[index] = iDanceDataArr[i];
			 index++;
		 }
		 
		 Arrays.sort(googlers);
		 
		 for (int i = 0; i < googlers.length; i++ )
		 {
			  totalScore = googlers[i];
			  if(isEligible(totalScore))
			  {
				  scoreSplit= split(totalScore);
				  if(isEligibleAfterConversion(scoreSplit))
				  {
					  eligibleGooglers++;
					  CommonUtility.printWithoutExtrLine(scoreSplit);
					  System.out.println("Eligible Googler #" + eligibleGooglers);
				  }
			  }
		 }

		 System.out.println("Output: " + strReturnValue + eligibleGooglers);
		 
		return strReturnValue + eligibleGooglers;
	}
	
	private boolean isEligibleAfterConversion(int[] scoreSplit)
	{
		boolean bReturn = false;
		
		if(currentNoOfSurprises == noOfSurprises)
		{
			return true;
		}
		
		int surpriseScore[]=  convertToSurpriseScore(scoreSplit);
		
		for(int i=0;i<surpriseScore.length;i++)
		{
			if(surpriseScore[i] >= bestScore)
			{
				bReturn = true;
				break;
			}
		}
		
		return bReturn;
	}
	
	private int[] convertToSurpriseScore(int[] scoreSplit)
	{
		int diff = scoreSplit[1] - scoreSplit[0];
		int diff1 = scoreSplit[2] - scoreSplit[1];
		int diff2 = scoreSplit[2] - scoreSplit[0];
		
		String strDiff = diff+""+diff1+""+diff2;
		
		//System.out.println("strDiff: " + strDiff);
		
		if (strDiff.equals("000"))
		{
			if ((scoreSplit[2] - 1) >= 0)
			{
				scoreSplit[0] = scoreSplit[0] + 1;
				scoreSplit[2] = scoreSplit[2] - 1;
			}
		}
		else if(strDiff.equals("011"))
		{
			if((scoreSplit[0] - 1) >= 0)
			{			
				scoreSplit[0] = scoreSplit[0] - 1;
				scoreSplit[1] = scoreSplit[1] + 1;
			}
		}
		else if(strDiff.equals("101"))
		{
			if((scoreSplit[1] - 1) >= 0)
			{			
				scoreSplit[1] = scoreSplit[1] - 1;
				scoreSplit[2] = scoreSplit[2] + 1;
			}
		}
		
		Arrays.sort(scoreSplit);
		
		if(isSurpriseScore(scoreSplit))
		{
			currentNoOfSurprises ++;
			CommonUtility.printWithoutExtrLine(scoreSplit);
			System.out.println(" Surprise#: "+currentNoOfSurprises);
			
		}

		return scoreSplit;
	}

	private boolean isSurpriseScore(int[] scoreSplit)
	{
		int diff = scoreSplit[1] - scoreSplit[0];
		int diff1 = scoreSplit[2] - scoreSplit[1];
		int diff2 = scoreSplit[2] - scoreSplit[0];
		
	//	System.out.println("isSurprise diff" + diff + " " + diff1 + " "+ diff2);
		
		return ((diff + diff1 + diff2) == 4);
		
	}
	
	private static int [] split(int totalScore)
	{
		int avgScore = totalScore / 3;
		int remScoreAvg = (totalScore - avgScore) / 2;
		
		int remScore = remScoreAvg;
		int remScore1 = remScoreAvg;
		int iReturn[] = new int[3];
		int mod = (totalScore - avgScore) % 2;
		
		if(mod == 1)
		{
			remScore++;
		}
		
		iReturn[0] = avgScore;
		iReturn[1] = remScore;
		iReturn[2] = remScore1;
		
		Arrays.sort(iReturn);
		
		CommonUtility.printWithoutExtrLine(iReturn);
		//System.out.println(" : Total Score :" + totalScore);
		
		return iReturn;
	}
	
	/*private boolean isSurprisingScore(int totalScore)
	{
		boolean bReturn = false;
		
		int bs1 = bestScore;
		int bs2 = bestScore -1;
		int bs3 = bestScore -2;
		
		if(bs1 >=0 && bs2 >=0 && bs3>=0 )
		{
		bReturn = (bestScore + ( bestScore -1 ) + (bestScore -2)) == totalScore;
		if(bReturn)
			System.out.println("Eligible Googler : [" + bestScore +", "+ ( bestScore -1 )+", " + (bestScore -2) +"] = " + totalScore + " (Surprise#" + (currentNoOfSurprises + 1)+")" );
		
		if(!bReturn)
		{
			bReturn = (bestScore + (( bestScore -2 ) * 2)) == totalScore;
			if (bReturn)
			System.out.println("Eligible Googler : [" + bestScore +", "+ ( bestScore -2 )+", " + (bestScore -2) +"] = " + totalScore + " (Surprise#" + (currentNoOfSurprises + 1)+")" );
		}
		}
		
		int bs4 = bestScore;
		int bs5 = bestScore +1;
		int bs6 = bestScore +2;
		
		if(bs1 <=10 && bs2 <=10 && bs3<=10 && !bReturn)
		{
			bReturn = (bestScore + ( bestScore +1 ) + (bestScore +2)) == totalScore;
			if(bReturn)
				System.out.println("Eligible Googler : [" + bestScore +", "+ ( bestScore +1 )+", " + (bestScore +2) +"] = " + totalScore + " (Surprise#" + (currentNoOfSurprises + 1)+")" );
			
			if(!bReturn)
			{
				bReturn = (bestScore + (( bestScore +2 ) * 2)) == totalScore;
				if (bReturn)
				System.out.println("Eligible Googler : [" + bestScore +", "+ ( bestScore +2 )+", " + (bestScore +2) +"] = " + totalScore + " (Surprise#" + (currentNoOfSurprises + 1)+")" );
			}
		}
		
		if (bReturn)
		{
			currentNoOfSurprises++;
		}
		return bReturn;
	}
*/
	private boolean isEligible(int totalScore)
	{
		boolean bReturn = false;
		
		if(noOfSurprises == 0 || currentNoOfSurprises >= noOfSurprises)
		{
		//	System.out.println("isEligibleIn: " + (bestScore + ((bestScore -1) * 2)) + " : " + totalScore);
			bReturn = (bestScore + ((bestScore -1) * 2)) <= totalScore;
		}
		else
		{
		//	System.out.println("isEligibleOut: " + (bestScore + ((bestScore -2) * 2)) + " : " + totalScore);
			bReturn = (bestScore + ((bestScore -2) * 2)) <= totalScore;
		}
		
		//System.out.println("isEligible: " + bReturn);
		
		return bReturn;
	}
	
}
