#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,613 ms
コンパイル使用メモリ	78,384 KB
実行使用メモリ	115,244 KB
最終ジャッジ日時	2024-11-21 13:52:53
合計ジャッジ時間	117,444 ms
ジャッジサーバーID （参考情報）	judge1 / judge5

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	55 ms 42,112 KB
testcase_01	AC	55 ms 82,184 KB
testcase_02	AC	56 ms 41,640 KB
testcase_03	AC	58 ms 82,152 KB
testcase_04	AC	58 ms 41,992 KB
testcase_05	AC	59 ms 82,220 KB
testcase_06	AC	60 ms 42,368 KB
testcase_07	AC	62 ms 82,008 KB
testcase_08	AC	66 ms 42,268 KB
testcase_09	AC	66 ms 82,364 KB
testcase_10	AC	67 ms 42,524 KB
testcase_11	AC	67 ms 82,876 KB
testcase_12	AC	81 ms 42,892 KB
testcase_13	AC	81 ms 82,860 KB
testcase_14	AC	84 ms 42,992 KB
testcase_15	AC	82 ms 82,512 KB
testcase_16	AC	83 ms 42,820 KB
testcase_17	AC	90 ms 83,568 KB
testcase_18	AC	85 ms 43,088 KB
testcase_19	AC	85 ms 83,472 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	TLE	-
testcase_42	TLE	-
testcase_43	TLE	-
testcase_44	TLE	-
testcase_45	TLE	-
testcase_46	TLE	-
testcase_47	TLE	-
testcase_48	TLE	-
testcase_49	TLE	-
testcase_50	TLE	-
testcase_51	TLE	-
testcase_52	TLE	-
testcase_53	TLE	-
testcase_54	TLE	-
testcase_55	TLE	-
testcase_56	TLE	-
testcase_57	TLE	-
testcase_58	TLE	-
testcase_59	TLE	-
testcase_60	WA	-
testcase_61	WA	-
testcase_62	WA	-
testcase_63	WA	-
testcase_64	WA	-
testcase_65	WA	-
testcase_66	WA	-
testcase_67	WA	-
testcase_68	WA	-
testcase_69	WA	-
testcase_70	WA	-
testcase_71	WA	-
testcase_72	WA	-
testcase_73	WA	-
testcase_74	WA	-
testcase_75	WA	-
testcase_76	WA	-
testcase_77	WA	-
testcase_78	WA	-
testcase_79	WA	-

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

ソースコード

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

結果

テストケース

ソースコード