#211148 (Java21) No.578 3 x N グリッド上のサイクルのサイズ(easy)

提出ソース

結果

問題	No.578 3 x N グリッド上のサイクルのサイズ(easy)
ユーザー	37zigen
提出日時	2017-10-22 19:59:45
言語	Java21 (openjdk 21)
結果	AC
実行時間	91 ms / 2,000 ms
コード長	4,460 bytes
コンパイル時間	2,153 ms
コンパイル使用メモリ	74,548 KB
実行使用メモリ	52,288 KB
最終ジャッジ日時	2023-08-13 17:40:30
合計ジャッジ時間	7,548 ms
ジャッジサーバーID （参考情報）	judge13 / judge15

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	43 ms 49,232 KB
testcase_01	AC	43 ms 49,092 KB
testcase_02	AC	44 ms 49,112 KB
testcase_03	AC	44 ms 49,416 KB
testcase_04	AC	46 ms 49,424 KB
testcase_05	AC	46 ms 49,340 KB
testcase_06	AC	47 ms 49,380 KB
testcase_07	AC	48 ms 49,212 KB
testcase_08	AC	49 ms 49,380 KB
testcase_09	AC	51 ms 49,164 KB
testcase_10	AC	52 ms 49,368 KB
testcase_11	AC	59 ms 51,004 KB
testcase_12	AC	60 ms 50,972 KB
testcase_13	AC	60 ms 50,708 KB
testcase_14	AC	60 ms 50,968 KB
testcase_15	AC	71 ms 50,756 KB
testcase_16	AC	61 ms 50,876 KB
testcase_17	AC	74 ms 51,408 KB
testcase_18	AC	74 ms 51,396 KB
testcase_19	AC	75 ms 51,424 KB
testcase_20	AC	77 ms 51,288 KB
testcase_21	AC	80 ms 51,428 KB
testcase_22	AC	80 ms 51,196 KB
testcase_23	AC	80 ms 51,240 KB
testcase_24	AC	81 ms 51,368 KB
testcase_25	AC	84 ms 51,324 KB
testcase_26	AC	77 ms 51,308 KB
testcase_27	AC	78 ms 51,632 KB
testcase_28	AC	77 ms 51,780 KB
testcase_29	AC	77 ms 49,896 KB
testcase_30	AC	79 ms 51,772 KB
testcase_31	AC	78 ms 51,860 KB
testcase_32	AC	81 ms 51,908 KB
testcase_33	AC	84 ms 51,744 KB
testcase_34	AC	87 ms 49,800 KB
testcase_35	AC	87 ms 51,876 KB
testcase_36	AC	90 ms 51,956 KB
testcase_37	AC	88 ms 52,152 KB
testcase_38	AC	88 ms 51,720 KB
testcase_39	AC	86 ms 52,288 KB
testcase_40	AC	87 ms 52,120 KB
testcase_41	AC	89 ms 50,196 KB
testcase_42	AC	89 ms 51,732 KB
testcase_43	AC	88 ms 51,948 KB
testcase_44	AC	85 ms 51,876 KB
testcase_45	AC	86 ms 52,120 KB
testcase_46	AC	89 ms 51,660 KB
testcase_47	AC	91 ms 51,888 KB
testcase_48	AC	88 ms 51,768 KB
testcase_49	AC	91 ms 51,908 KB

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

ソースコード

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

結果

テストケース

ソースコード