#211147 (Java21) No.579 3 x N グリッド上のサイクルのサイズ(hard)

提出ソース

結果

問題	No.579 3 x N グリッド上のサイクルのサイズ(hard)
ユーザー	37zigen
提出日時	2017-10-22 19:59:23
言語	Java21 (openjdk 21)
結果	TLE
実行時間	-
コード長	4,460 bytes
コンパイル時間	2,112 ms
コンパイル使用メモリ	74,896 KB
実行使用メモリ	59,348 KB
最終ジャッジ日時	2023-08-13 17:39:58
合計ジャッジ時間	57,212 ms
ジャッジサーバーID （参考情報）	judge13 / judge11

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	42 ms 53,600 KB
testcase_01	AC	42 ms 49,140 KB
testcase_02	AC	43 ms 49,284 KB
testcase_03	AC	43 ms 49,232 KB
testcase_04	AC	44 ms 49,276 KB
testcase_05	AC	46 ms 49,280 KB
testcase_06	AC	47 ms 49,160 KB
testcase_07	AC	47 ms 49,224 KB
testcase_08	AC	49 ms 49,228 KB
testcase_09	AC	51 ms 49,088 KB
testcase_10	AC	53 ms 49,308 KB
testcase_11	AC	57 ms 50,952 KB
testcase_12	AC	69 ms 50,692 KB
testcase_13	AC	58 ms 51,208 KB
testcase_14	AC	70 ms 50,744 KB
testcase_15	AC	71 ms 50,812 KB
testcase_16	AC	61 ms 50,712 KB
testcase_17	AC	73 ms 51,564 KB
testcase_18	AC	74 ms 51,576 KB
testcase_19	AC	75 ms 51,400 KB
testcase_20	TLE	-
testcase_21	TLE	-
testcase_22	TLE	-
testcase_23	TLE	-
testcase_24	TLE	-
testcase_25	TLE	-
testcase_26	TLE	-
testcase_27	TLE	-
testcase_28	TLE	-
testcase_29	TLE	-
testcase_30	TLE	-
testcase_31	TLE	-
testcase_32	TLE	-
testcase_33	TLE	-
testcase_34	TLE	-
testcase_35	TLE	-
testcase_36	TLE	-
testcase_37	TLE	-
testcase_38	TLE	-
testcase_39	TLE	-
testcase_40	TLE	-
testcase_41	--	-
testcase_42	--	-
testcase_43	--	-
testcase_44	--	-
testcase_45	--	-
testcase_46	--	-
testcase_47	--	-
testcase_48	--	-
testcase_49	--	-
testcase_50	--	-
testcase_51	--	-
testcase_52	--	-
testcase_53	--	-
testcase_54	--	-
testcase_55	--	-
testcase_56	--	-
testcase_57	--	-
testcase_58	--	-
testcase_59	--	-
testcase_60	--	-
testcase_61	--	-
testcase_62	--	-
testcase_63	--	-
testcase_64	--	-
testcase_65	--	-
testcase_66	--	-
testcase_67	--	-
testcase_68	--	-
testcase_69	--	-
testcase_70	--	-
testcase_71	--	-
testcase_72	--	-
testcase_73	--	-
testcase_74	--	-
testcase_75	--	-
testcase_76	--	-
testcase_77	--	-
testcase_78	--	-
testcase_79	--	-

権限があれば一括ダウンロードができます

ソースコード

raw source code

import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayDeque;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.NoSuchElementException;

public class Main {
	public static void main(String[] args) {
		// long t = System.currentTimeMillis();
		new Main().run();
		// System.err.println(System.currentTimeMillis() - t);
	}

	final long MOD = 1_000_000_000 + 7;

	void run() {
		Scanner sc = new Scanner();
		int n = sc.nextInt();
		String[] s = new String[] { "0000000000", "0034534306", "1023400005", "2032334000", "3043243450", "1003423000",
				"2004332300", "1000403205", "2030000340", "0000300004" };
		long[][] mat1 = new long[10][10];
		long[][] mat2 = new long[10][10];
		for (int i = 0; i < 10; ++i) {
			for (int j = 0; j < 10; ++j) {
				mat1[i][j] = (int) (s[i].charAt(j) - '0');
				if (mat1[i][j] > 0)
					++mat2[i][j];
			}
		}
		++mat2[0][0];
		++mat2[1][1];

		long[][] x = new long[10][10];
		long[][] v = new long[10][1];
		v[0][0] = 1;
		for (int i = 0; i <= n; ++i) {
			x = add(x, mul(mul(pow(mat2, i), mat1), pow(mat2, n - i)));
		}
		v = mul(x, v);
		System.out.println(v[1][0]);
	}

	long[][] pow(long[][] a, long n) {
		long[][] ret = new long[a.length][a.length];
		for (int i = 0; i < ret.length; ++i)
			ret[i][i] = 1;
		long[][] p = new long[a.length][a.length];
		for (int i = 0; i < a.length; ++i) {
			for (int j = 0; j < a[i].length; ++j) {
				p[i][j] = a[i][j];
			}
		}
		for (; n > 0; n >>= 1, p = mul(p, p)) {
			if (n % 2 == 1)
				ret = mul(ret, p);
		}
		return ret;
	}

	long[][] mul(long[][] a, long[][] b) {
		long[][] ret = new long[a.length][b[0].length];
		for (int i = 0; i < a.length; ++i) {
			for (int j = 0; j < b[i].length; ++j) {
				for (int k = 0; k < a[i].length; ++k) {
					ret[i][j] = (ret[i][j] + a[i][k] * b[k][j] % MOD) % MOD;
				}
			}
		}
		return ret;
	}

	long[][] add(long[][] a, long[][] b) {
		if (a.length != b.length || a[0].length != b[0].length)
			throw new AssertionError();
		long[][] ret = new long[a.length][a[0].length];
		for (int i = 0; i < a.length; ++i) {
			for (int j = 0; j < a[i].length; ++j) {
				ret[i][j] = (a[i][j] + b[i][j]) % MOD;
			}
		}
		return ret;
	}

	static void tr(Object... objects) {
		System.out.println(Arrays.deepToString(objects));
	}

	class Scanner {
		private final InputStream in = System.in;
		private final byte[] buffer = new byte[1024];
		private int ptr = 0;
		private int buflen = 0;

		private boolean hasNextByte() {
			if (ptr < buflen) {
				return true;
			} else {
				ptr = 0;
				try {
					buflen = in.read(buffer);
				} catch (IOException e) {
					e.printStackTrace();
				}
				if (buflen <= 0) {
					return false;
				}
			}
			return true;
		}

		private int readByte() {
			if (hasNextByte())
				return buffer[ptr++];
			else
				return -1;
		}

		private boolean isPrintableChar(int c) {
			return 33 <= c && c <= 126;
		}

		private void skipUnprintable() {
			while (hasNextByte() && !isPrintableChar(buffer[ptr]))
				ptr++;
		}

		public boolean hasNext() {
			skipUnprintable();
			return hasNextByte();
		}

		public String next() {
			if (!hasNext())
				throw new NoSuchElementException();
			StringBuilder sb = new StringBuilder();
			int b = readByte();
			while (isPrintableChar(b)) {
				sb.appendCodePoint(b);
				b = readByte();
			}
			return sb.toString();
		}

		public int nextInt() {
			if (!hasNext())
				throw new NoSuchElementException();
			int n = 0;
			boolean minus = false;
			int b = readByte();
			if (b == '-') {
				minus = true;
				b = readByte();
			}
			if (b < '0' || '9' < b) {
				throw new NumberFormatException();
			}
			while (true) {
				if ('0' <= b && b <= '9') {
					n *= 10;
					n += b - '0';
				} else if (b == -1 || !isPrintableChar(b)) {
					return minus ? -n : n;
				} else {
					throw new NumberFormatException();
				}
				b = readByte();
			}
		}

		public long nextLong() {
			if (!hasNext())
				throw new NoSuchElementException();
			long n = 0;
			boolean minus = false;
			int b = readByte();
			if (b == '-') {
				minus = true;
				b = readByte();
			}
			if (b < '0' || '9' < b) {
				throw new NumberFormatException();
			}
			while (true) {
				if ('0' <= b && b <= '9') {
					n *= 10;
					n += b - '0';
				} else if (b == -1 || !isPrintableChar(b)) {
					return minus ? -n : n;
				} else {
					throw new NumberFormatException();
				}
				b = readByte();
			}
		}
	}
}

yukicoder

結果

テストケース

ソースコード