Tontti

import java.util.*;
import java.math.*;
 
public class Tehtava3 {
	
	private static int[][] arvot;
	
	public static void laskeTavat(int korkeus, int leveys, String[] metsa, int puita) {
		
		
		//long t = System.currentTimeMillis();
		int tulos = 0;
		
		
		int pieninKoko = (int)Math.ceil(Math.sqrt(puita));
		int maksimiKoko = Math.min(korkeus, leveys);
		
 
		
	//	System.out.println("1. Osa: " + (System.currentTimeMillis()-t) + "ms");
		
		
		int maara = 0;
		int maaraAlueella = 0;
			
			for (int j = 0; j<=leveys-pieninKoko; j++) {
				
				for (int k = 0; k<=korkeus-pieninKoko; k++) {
					
					for (int i = pieninKoko; i<=maksimiKoko; i++) {
						if (k+i > korkeus || j+i > leveys)
							break;
						
						maaraAlueella = puidenMaara(j, k, i);
						if (maaraAlueella == puita) {
						maara++;
						}
						else if (maaraAlueella > puita) {
							break;
						}
					
					}
					
				}
				
			}
		
		//System.out.println("2. Osa: " + (System.currentTimeMillis()-t) + "ms");
		System.out.println(maara);
		
	}
	
 
 
	
	public static int puidenMaara(int alkux, int alkuy, int koko) {
		
		
		int arvo1 = 0;
		int arvo2 = 0;
		int arvo3 = 0;
		int arvo4 = arvot[alkux+koko-1][alkuy+koko-1];;
		
		if (alkux != 0) {
				arvo3 = arvot[alkux-1][alkuy+koko-1];
		}
		
		
		if (alkuy != 0) {
			arvo2 = arvot[alkux+koko-1][alkuy-1];
			
			if (alkux != 0)
				arvo1 = arvot[alkux-1][alkuy-1];
		}
		
		return (arvo1+arvo4-arvo2-arvo3);
	}
	
 
	
	public static String[] luoSatunnainen(int leveys, int korkeus) {
		StringBuilder sb;
		String[] metsa = new String[korkeus];
		Random r = new Random();
		for (int i = 0; i<korkeus; i++) {
			sb = new StringBuilder();
			for (int j = 0; j<leveys; j++) {
				int a = r.nextInt(9);
				if (a <= 1)
					sb.append('*');
				else
					sb.append('.');
				
			}
			
			metsa[i] = sb.toString();
		}
		
		return metsa;
	}
	
	
	public static void tulostaMetsa(String[] metsa) {
	
		for (int i = 0; i<metsa.length; i++) {
			for (int j = 0; j<metsa[0].length(); j++) {
				System.out.print(metsa[i].charAt(j) + " ");
				
			}
			System.out.println();
			System.out.println("--------------------------------------");
		}
	}
	
	public static void luoArvoTaulukko(String[] metsa) {
		// metsa[0] = leveys
		//metsa = korkeus
		
		//arvot[0] = korkeus
		// arvot = leveys
		
		//long t1 = System.currentTimeMillis();
		
		int summa1 = 0;
		for (int i = 0; i<metsa[0].length(); i++) {
			if (metsa[0].charAt(i) == '*') {
				summa1++;
			}
			arvot[i][0] = summa1;
		}
		
		int summa2 = 0;
		for (int y = 1; y<metsa.length; y++) {
			summa1 = 0;
			summa2 = 0;
			
			for (int x = 0; x<metsa[0].length(); x++) {
				
				if (metsa[y].charAt(x) == '*')
					summa2++;
				
				summa1 = arvot[x][y-1];
						
				
				arvot[x][y] = (summa1+summa2);
			}
		}
		
		//System.out.println("arvotaulukon luontiin meni " + (System.currentTimeMillis()-t1) + "ms");
	}
	
	public static void tulostaArvoTaulukko() {
		System.out.println();
		for (int i = 0; i<arvot[0].length; i++) {
			for (int j = 0; j<arvot.length; j++) {
				System.out.print(arvot[j][i] + " | ");
			}
			System.out.println("--------------");
		}
	}
 
	public static void main(String[] args) {
		
		
		//IO io = new IO();
		//int korkeus = io.nextInt();
		//int leveys = io.nextInt();
		//int puita = io.nextInt();
		
 
		
		int korkeus = 1500;
		int leveys = 1500;
		int puita = 3000;
		
		
		String[] metsa = new String[korkeus];
		
		//kokaika = 0;
		
	//	for (int i = 0; i<korkeus; i++) {
	//		metsa[i] = io.next();
	//	}
		
		arvot = new int[leveys][korkeus];
		
 
		//metsa[0] = "..**..";
		//metsa[1] = "**....";
		//metsa[2] = "*...*.";
		//metsa[3] = "..*...";
		
		
		
		
		metsa = luoSatunnainen(leveys, korkeus);
	//	tulostaMetsa(metsa);
		
 
	//
		luoArvoTaulukko(metsa);	
		
		//tulostaArvoTaulukko();
		laskeTavat(korkeus, leveys, metsa, puita);
 
	}
	
 
}

Test 1

Group: 1

Verdict: TIME LIMIT EXCEEDED

Test 2

Group: 1

Verdict: TIME LIMIT EXCEEDED

Test 3

Group: 1

Verdict: TIME LIMIT EXCEEDED

Test 4

Group: 1

Verdict: TIME LIMIT EXCEEDED

Test 5

Group: 1

Verdict: TIME LIMIT EXCEEDED

Test 6

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 7

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 8

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 9

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 10

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 11

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 12

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 13

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 14

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 15

Group: 3

Verdict: TIME LIMIT EXCEEDED