結果

問題 No.866 レベルKの正方形
ユーザー Grenache
提出日時 2019-08-20 13:28:25
言語 Java8
(openjdk 1.8.0.222)
結果
TLE  
実行時間 -
コード長 6,559 Byte
コンパイル時間 2,732 ms
使用メモリ 437,964 KB
最終ジャッジ日時 2019-08-20 13:29:51

テストケース

テストケース表示
入力 結果 実行時間
使用メモリ
in01.txt AC 77 ms
19,264 KB
in02.txt AC 78 ms
19,260 KB
in03.txt AC 77 ms
19,268 KB
in04.txt AC 76 ms
19,268 KB
in05.txt AC 77 ms
19,260 KB
in06.txt AC 75 ms
19,268 KB
in07.txt AC 75 ms
19,264 KB
in08.txt AC 76 ms
19,268 KB
in09.txt AC 5,996 ms
437,960 KB
in10.txt AC 5,815 ms
437,956 KB
in11.txt AC 5,823 ms
437,960 KB
in12.txt AC 5,923 ms
437,964 KB
in13.txt TLE -
in14.txt TLE -
in15.txt TLE -
in16.txt AC 5,936 ms
437,964 KB
in17.txt TLE -
in18.txt AC 5,967 ms
437,960 KB
in19.txt AC 5,811 ms
437,960 KB
in20.txt TLE -
in21.txt AC 5,757 ms
437,960 KB
in22.txt AC 3,885 ms
437,956 KB
sample_01.txt AC 74 ms
19,252 KB
sample_02.txt AC 74 ms
19,252 KB
sample_03.txt AC 82 ms
20,148 KB
テストケース一括ダウンロード

ソースコード

diff #
import java.io.*;
import java.util.*;


public class Main_yukicoder866_2 {

	private static Scanner sc;
	private static Printer pr;

	private static void solve() {
		int h = sc.nextInt();
		int w = sc.nextInt();
		int k = sc.nextInt();

		char[][] c = new char[h][];
		for (int i = 0; i < h; i++) {
			c[i] = sc.nextCharArray(w);
		}

		int[][][] dp = new int[26][h][w];
		final int INF = Integer.MAX_VALUE / 2;
		for (char cc = 'a'; cc <= 'z'; cc++) {
			int cci = cc - 'a';
			
			for (int i = 0; i < h; i++) {
				Arrays.fill(dp[cci][i], INF);
			}
			
			for (int i = 0; i < h; i++) {
				for (int j = 0; j < w; j++) {
					if (c[i][j] == cc) {
						dp[cci][i][j] = 1;
					}
				}
			}

			for (int i = h - 1; i >= 0; i--) {
				for (int j = w - 1; j >= 0; j--) {
					int max = Math.min(h - i, w - j);
					int tmp = dp[cci][i][j];
					if (i + 1 < h && j + 1 < w) {
						tmp = Math.min(tmp, dp[cci][i + 1][j + 1] + 1);
					}
					if (i + 1 < h) {
						tmp = Math.min(tmp, dp[cci][i + 1][j] + 1);
					}
					if (j + 1 < w) {
						tmp = Math.min(tmp, dp[cci][i][j + 1] + 1);
					}
					if (tmp > max) {
						tmp = INF;
					}
					dp[cci][i][j] = tmp;
				}
			}
		}

		long ans = 0;
		int[] tmp = new int[27];
		for (int i = 0; i < h; i++) {
			for (int j = 0; j < w; j++) {
				int max = Math.min(h - i, w - j);
				tmp[26] = max + 1;;
				for (int l = 0; l < 26; l++) {
					if (dp[l][i][j] == INF) {
						tmp[l] = max + 1;
					} else {
						tmp[l] = dp[l][i][j];
					}
				}
				Arrays.sort(tmp);

				ans += tmp[k] - tmp[k - 1];
			}
		}
		
		pr.println(ans);
	}

	// ---------------------------------------------------
	public static void main(String[] args) {
		sc = new Scanner(System.in);
		pr = new Printer(System.out);
			
		solve();
			
		pr.close();
		sc.close();
	}

	static class Scanner {
		BufferedReader br;

		Scanner (InputStream in) {
			br = new BufferedReader(new InputStreamReader(in));
		}

		private boolean isPrintable(int ch) {
			return ch >= '!' && ch <= '~';
		}

		private boolean isCRLF(int ch) {
			return ch == '\n' || ch == '\r' || ch == -1;
		}

		private int nextPrintable() {
			try {
				int ch;
				while (!isPrintable(ch = br.read())) {
					if (ch == -1) {
						throw new NoSuchElementException();
					}
				}

				return ch;
			} catch (IOException e) {
				throw new NoSuchElementException();
			}
		}

		String next() {
			try {
				int ch = nextPrintable();
				StringBuilder sb = new StringBuilder();
				do {
					sb.appendCodePoint(ch);
				} while (isPrintable(ch = br.read()));

				return sb.toString();
			} catch (IOException e) {
				throw new NoSuchElementException();
			}
		}

		int nextInt() {
			try {
				// parseInt from Integer.parseInt()
				boolean negative = false;
				int res = 0;
				int limit = -Integer.MAX_VALUE;
				int radix = 10;

				int fc = nextPrintable();
				if (fc < '0') {
					if (fc == '-') {
						negative = true;
						limit = Integer.MIN_VALUE;
					} else if (fc != '+') {
						throw new NumberFormatException();
					}
					fc = br.read();
				}
				int multmin = limit / radix;

				int ch = fc;
				do {
					int digit = ch - '0';
					if (digit < 0 || digit >= radix) {
						throw new NumberFormatException();
					}
					if (res < multmin) {
						throw new NumberFormatException();
					}
					res *= radix;
					if (res < limit + digit) {
						throw new NumberFormatException();
					}
					res -= digit;

				} while (isPrintable(ch = br.read()));

				return negative ? res : -res;
			} catch (IOException e) {
				throw new NoSuchElementException();
			}
		}

		long nextLong() {
			try {
				// parseLong from Long.parseLong()
				boolean negative = false;
				long res = 0;
				long limit = -Long.MAX_VALUE;
				int radix = 10;

				int fc = nextPrintable();
				if (fc < '0') {
					if (fc == '-') {
						negative = true;
						limit = Long.MIN_VALUE;
					} else if (fc != '+') {
						throw new NumberFormatException();
					}
					fc = br.read();
				}
				long multmin = limit / radix;

				int ch = fc;
				do {
					int digit = ch - '0';
					if (digit < 0 || digit >= radix) {
						throw new NumberFormatException();
					}
					if (res < multmin) {
						throw new NumberFormatException();
					}
					res *= radix;
					if (res < limit + digit) {
						throw new NumberFormatException();
					}
					res -= digit;

				} while (isPrintable(ch = br.read()));

				return negative ? res : -res;
			} catch (IOException e) {
				throw new NoSuchElementException();
			}
		}

		float nextFloat() {
			return Float.parseFloat(next());
		}

		double nextDouble() {
			return Double.parseDouble(next());
		}

		String nextLine() {
			try {
				int ch;
				while (isCRLF(ch = br.read())) {
					if (ch == -1) {
						throw new NoSuchElementException();
					}
				}
				StringBuilder sb = new StringBuilder();
				do {
					sb.appendCodePoint(ch);
				} while (!isCRLF(ch = br.read()));

				return sb.toString();
			} catch (IOException e) {
				throw new NoSuchElementException();
			}
		}
		
		char[] nextCharArray(int n) {
			char[] ret = new char[n];
			try {
				int ch = nextPrintable();
				int i = 0;
				do {
					ret[i++] = (char)ch;
				} while (isPrintable(ch = br.read()));

				return ret;
			} catch (IOException e) {
				throw new NoSuchElementException();
			}
		}

		int[] nextIntArray(int n) {
			int[] ret = new int[n];
			for (int i = 0; i < n; i++) {
				ret[i] = sc.nextInt();
			}
			
			return ret;
		}

		int[][] nextIntArrays(int n, int m) {
			int[][] ret = new int[m][n];
			for (int i = 0; i < n; i++) {
				for (int j = 0; j < m; j++) {
					ret[j][i] = sc.nextInt();
				}
			}
			
			return ret;
		}

		void close() {
			try {
				br.close();
			} catch (IOException e) {
//				throw new NoSuchElementException();
			}
		}
	}

	static class Printer extends PrintWriter {
		Printer(OutputStream out) {
			super(out);
		}
		
		void printInts(int... a) {
			StringBuilder sb = new StringBuilder(32);
			for (int i = 0, size = a.length; i < size; i++) {
				if (i > 0) {
					sb.append(' ');
				}
				sb.append(a[i]);
			}

			println(sb);
		}
		
		void printLongs(long... a) {
			StringBuilder sb = new StringBuilder(64);
			for (int i = 0, size = a.length; i < size; i++) {
				if (i > 0) {
					sb.append(' ');
				}
				sb.append(a[i]);
			}

			println(sb);
		}
		
		void printStrings(String... a) {
			StringBuilder sb = new StringBuilder(32);
			for (int i = 0, size = a.length; i < size; i++) {
				if (i > 0) {
					sb.append(' ');
				}
				sb.append(a[i]);
			}

			println(sb);
		}
	}
}
0