Tontti

/**
 * Created by Frans on 5.10.2015.
 */
public class Tontti{

    static int height;
    static int width;
    static int[][] kenttaSumma;
    static int wantedTrees;
    //static long runAmount;

    public static void main(String[] args){

        /*//todo
        int futureWidth = 1750;
        int futureHeight = 1750;
        int futurewantedTrees = (int) (Math.random() * futureHeight);
        height = futureHeight;
        width = futureWidth;
        wantedTrees = futurewantedTrees;

        kenttaSumma = new int[width+1][height+1];

        String[] lines = new String[height+1];
        for (int y=1; y<=height; y++) {
            lines[y] = "";
            for (int x2=1; x2<=width; x2++){
                if (Math.random() > 0.3){
                    lines[y] = lines[y] + ".";
                }else {
                    lines[y] = lines[y] + "*";
                }
            }
        }*/


        IO io = new IO();

        height = io.nextInt();
        width = io.nextInt();
        wantedTrees = io.nextInt();


        //long aika = System.nanoTime();


        kenttaSumma = new int[width+1][height+1];

        int foundArea = 0;
        int minSize = (int) (Math.sqrt(wantedTrees) + 0.99);

        //calculate trees and save it in to array and summed area table
        mainLoop:
        {
            for (int y = 1; y <= height; y++) {
                String line = io.next();
                //String line = lines[y];

                int sumRow = 0;
                for (int x = 1; x <= width; x++) {
                    if (line.charAt(x - 1) == '*') {
                        sumRow++;
                    }
                    kenttaSumma[x][y] = sumRow + kenttaSumma[x][y - 1];

                    int maxSize = min(x, y);

                    //todo remove
                    if (x==1950 && y==1950){
                        //stop here
                        break mainLoop;
                    }

                    if (maxSize >= minSize) {
                        foundArea += countAreasB(maxSize, minSize, x, y, 0);
                    }

                }
            }
        }

        io.println(foundArea);

        /*aika = (System.nanoTime()-aika)/1000000;
        io.println("---");
        io.println("Time: " + aika);
        */
        //io.println("Loops: " + runAmount);


        io.close();
    }

    //rekursiivinen binaarihaku
    // x2 = x -1
    // y2 = y -1
    private static int countAreasB(int maxSize,int minSize, int x, int y, int prevSize){
        int areasFound = 0;
        int tSize = (int) ((maxSize + minSize)/2.0 + 0.99);

        //lopetaan kun sama on jo katsottu
        if (prevSize != tSize){
            prevSize = tSize;

            int sum = kenttaSumma[x][y]+kenttaSumma[x-tSize][y-tSize]-kenttaSumma[x][y-tSize]-kenttaSumma[x-tSize][y];

            if(sum > wantedTrees){
                tSize = (tSize - minSize == 1)?minSize:tSize;
                areasFound += countAreasB(tSize, minSize, x, y, prevSize);
            }else if (sum < wantedTrees){
                areasFound += countAreasB(maxSize, tSize, x, y, prevSize);
            }else {
                //molemmat suunnat ovat mahdollisia --> kokeillaan kaikki lahella olevat
                areasFound ++;
                areasFound += countAreasL(tSize+1, maxSize, x, y, 1);
                areasFound += countAreasL(tSize-1, minSize, x, y, -1);
            }
        }
        
        return areasFound;
    }

    //suoraviivainen haku
    // x2 = x-1
    // y2 = y-1
    private static int countAreasL(int startSize,int endSize, int x, int y, int direction){
        int areasFound = 0;

        for (int tSize=startSize; direction*tSize<=endSize*direction; tSize=tSize+direction){

            int sum = kenttaSumma[x][y]+kenttaSumma[x-tSize][y-tSize]-kenttaSumma[x][y-tSize]-kenttaSumma[x-tSize][y];
            if (sum == wantedTrees){
                areasFound ++;
            }else{
                break;
            }

        }

        return areasFound;
    }

    //just for lulz
    private static int max(final int a, final int b){
        return (a>b)?a:b;
    }
    private static int min(final int a, final int b){
        return (a<b)?a:b;
    }


}

Test 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 1

Verdict: ACCEPTED

Test 3

Group: 1

Verdict: ACCEPTED

Test 4

Group: 1

Verdict: ACCEPTED

Test 5

Group: 1

Verdict: ACCEPTED

Test 6

Group: 2

Verdict: ACCEPTED

Test 7

Group: 2

Verdict: ACCEPTED

Test 8

Group: 2

Verdict: ACCEPTED

Test 9

Group: 2

Verdict: ACCEPTED

Test 10

Group: 2

Verdict: ACCEPTED

Test 11

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 12

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 13

Group: 3

Verdict: WRONG ANSWER

Test 14

Group: 3

Verdict: WRONG ANSWER

Test 15

Group: 3

Verdict: WRONG ANSWER