結果

問題 No.435 占い(Extra)
ユーザー tanzakutanzaku
提出日時 2016-10-15 19:57:25
言語 Java21
(openjdk 21)
結果
AC  
実行時間 540 ms / 2,000 ms
コード長 8,871 bytes
コンパイル時間 3,403 ms
コンパイル使用メモリ 91,244 KB
実行使用メモリ 46,228 KB
最終ジャッジ日時 2024-10-08 12:02:13
合計ジャッジ時間 11,677 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 64 ms
37,384 KB
testcase_01 AC 64 ms
37,048 KB
testcase_02 AC 66 ms
37,200 KB
testcase_03 AC 67 ms
37,064 KB
testcase_04 AC 64 ms
37,372 KB
testcase_05 AC 65 ms
37,412 KB
testcase_06 AC 95 ms
37,988 KB
testcase_07 AC 64 ms
37,192 KB
testcase_08 AC 66 ms
37,148 KB
testcase_09 AC 67 ms
37,160 KB
testcase_10 AC 89 ms
38,016 KB
testcase_11 AC 94 ms
37,988 KB
testcase_12 AC 99 ms
38,312 KB
testcase_13 AC 118 ms
38,536 KB
testcase_14 AC 125 ms
38,400 KB
testcase_15 AC 158 ms
39,096 KB
testcase_16 AC 199 ms
39,724 KB
testcase_17 AC 320 ms
46,228 KB
testcase_18 AC 108 ms
38,212 KB
testcase_19 AC 123 ms
38,084 KB
testcase_20 AC 259 ms
38,508 KB
testcase_21 AC 267 ms
38,344 KB
testcase_22 AC 276 ms
38,576 KB
testcase_23 AC 309 ms
39,332 KB
testcase_24 AC 392 ms
40,740 KB
testcase_25 AC 540 ms
45,928 KB
testcase_26 AC 235 ms
38,400 KB
testcase_27 AC 259 ms
38,308 KB
testcase_28 AC 55 ms
37,216 KB
testcase_29 AC 255 ms
38,280 KB
testcase_30 AC 227 ms
38,356 KB
testcase_31 AC 235 ms
38,420 KB
testcase_32 AC 256 ms
38,576 KB
testcase_33 AC 351 ms
39,484 KB
testcase_34 AC 494 ms
44,080 KB
testcase_35 AC 311 ms
39,364 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.io.*;
import java.math.*;
import java.util.*;
import java.util.Map.Entry;

import static java.util.Arrays.*;

public class Main {
	private static final int mod = (int)1e9+7;

	final Random random = new Random(0);
	final IOFast io = new IOFast();

//	TreeMap<Integer, Integer>[] comb = new TreeMap[100000];
	
	/// MAIN CODE
	public void run() throws IOException {
//		dump(divisor);
//		for(int i = 2; i < 9; i++) {
//			List<Integer> xs = new ArrayList<>();
//			for(int j = 0, k = 1; j < 10; j++, k = k * i % 9) {
//				xs.add(k);
//			}
//			dump(xs);
//		}
		//*
		int TEST_CASE = Integer.parseInt(new String(io.nextLine()).trim());
//		int TEST_CASE = 1;
		while(TEST_CASE-- != 0) {
			int n = io.nextInt();
			int x = io.nextInt();
			int a = io.nextInt();
			int b = io.nextInt();
			int m = io.nextInt();
			io.out.println(fast(n, x, a, b, m));
		}
		//*/
	}
	
	
	static int[] inv = new int[9];
	static {
		for(int i = 1; i < inv.length; i++) if(i % 3 != 0) {
			inv[i] = (int)modInverse(i, 9);
		}
	}
	int fast(int n, int x, int a, int b, int m) {
		long sum = 0;
		int nonzero = 0;
		long comb = 1;
		int m3 = 0;
		for(int i = 0; i < n; i++) {
//			proc(mul, n - 1 - i, 1);
//			proc(mul, i + 1, -1);
			if(m3 <= 1) sum += x % 10 * (m3 == 0 ? comb : comb * 3);
//			dump(x, comb, m3);
			if(x % 10 != 0) nonzero = 1;
			x = ((x ^ a) + b) % m;
			if(i != n - 1) {
				int v = n - 1 - i;
				while(v % 3 == 0) { v /= 3; m3++; }
				comb = comb * v % 9;
			}
			{
				int v = i + 1;
				while(v % 3 == 0) { v /= 3; m3--; }
				comb = comb * inv[v % 9] % 9;
			}
		}
		sum += nonzero * 9;
		return (int)((sum - nonzero) % 9 + nonzero);
	}
	
	static long modInverse(final long n, final long m) {
		final long[] res = extGcd(n, m);
		return res[2] != 1 ? -1 : (m + res[0]) % m;
	}
	
	static long[] extGcd(final long a, final long b) {
		if (b == 0) {
			return new long[] { 1, 0, a };
		}
		final long[] res = extGcd(b, a % b);
		long y = res[0];
		final long x = res[1];
		y -= (a / b) * x;
		res[0] = x;
		res[1] = y;
		return res;
	}
	
	int[] loop = new int[]{
			-1, -1, 6, -1, 3, 6, -1, 3, 2
		};
	int[] mul = new int[9];
	short[] divisor;
	void proc(int[] mp, int x, int v) {
		for(; x != 0 && x != 1; x /= divisor[x]) {
			if(divisor[x] == -1) {
				mp[x % 9] += v;
				break;
			}
			mp[divisor[x] % 9] += v;
		}
	}
	int fast1(int n, int x, int a, int b, int m) {
		if(divisor == null) {
			divisor = minDivisor(10000000+1);
		}
		
		long sum = 0;
		int nonzero = 0;
		Arrays.fill(mul, 0);
		for(int i = 0; i < n; i++) {
			long xx;
			if(mul[3] >= 2) {
				xx = 0;
			} else {
				xx = 1;
				for(int j = 2; j < mul.length; j++) if(mul[j] != 0) {
					int v = mul[j];
					if(loop[j] != -1) v %= loop[j];
					xx = xx * powmod(j, v, 9);
				}
			}
			proc(mul, n - 1 - i, 1);
			proc(mul, i + 1, -1);
			sum += x % 10 * xx;
//			dump(x, xx);
			if(x % 10 != 0) nonzero = 1;
			x = ((x ^ a) + b) % m;
		}
		sum += nonzero * 9;
		return (int)((sum - nonzero) % 9 + nonzero);
	}
	
	private static short[] minDivisor(int n) {
		final short[] min = new short[n+1];
		min[0] = min[1] = 1;
		for(int i = 2; i <= n; i++) {
			if(min[i] == 0) {
				final short v = i >= Short.MAX_VALUE ? -1 : (short)i;
				min[i] = v;
				for(long j = (long)i*i; j <= n; j += i) {
					if(min[(int)j] == 0) {
						min[(int)j] = v;
					}
				}
			}
		}
		return min;
	}
	
	static long powmod(long n, long r, int m) {
		long res = 1;
		for(; r != 0; r >>>= 1, n = n * n % m) {
			if((r&1) == 1) {
				res = res * n;
				if(res >= m) {
					res %= m;
				}
			}
		}
		return res;
	}
	
	int slow(char[] cs) {
		for(int i = 0; i < cs.length; i++) cs[i] -= '0';
		for(int i = 0; i < cs.length; i++) {
			for(int j = 0; j < cs.length - 1 - i; j++) {
				cs[j] += cs[j+1];
				if(cs[j] >= 10) cs[j] -= 9;
			}
		}
		return cs[0];
	}
	
	/// TEMPLATE
	static int gcd(int n, int r) { return r == 0 ? n : gcd(r, n%r); }
	static long gcd(long n, long r) { return r == 0 ? n : gcd(r, n%r); }
	
	static <T> void swap(T[] x, int i, int j) { T t = x[i]; x[i] = x[j]; x[j] = t; }
	static void swap(int[] x, int i, int j) { int t = x[i]; x[i] = x[j]; x[j] = t; }

	void printArrayLn(int[] xs) { for(int i = 0; i < xs.length; i++) io.out.print(xs[i] + (i==xs.length-1?"\n":" ")); }
	void printArrayLn(long[] xs) { for(int i = 0; i < xs.length; i++) io.out.print(xs[i] + (i==xs.length-1?"\n":" ")); }
	
	static void dump(Object... o) { System.err.println(Arrays.deepToString(o)); } 
	
	void main() throws IOException {
		//		IOFast.setFileIO("rle-size.in", "rle-size.out");
		try { run(); }
		catch (EndOfFileRuntimeException e) { }
		io.out.flush();
	}
	public static void main(String[] args) throws IOException { new Main().main(); }
	
	static class EndOfFileRuntimeException extends RuntimeException {
		private static final long serialVersionUID = -8565341110209207657L; }

	static
	public class IOFast {
		private BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
		private PrintWriter out = new PrintWriter(System.out);

		void setFileIn(String ins) throws IOException { in.close(); in = new BufferedReader(new FileReader(ins)); }
		void setFileOut(String outs) throws IOException { out.flush(); out.close(); out = new PrintWriter(new FileWriter(outs)); }
		void setFileIO(String ins, String outs) throws IOException { setFileIn(ins); setFileOut(outs); }

		private static int pos, readLen;
		private static final char[] buffer = new char[1024 * 8];
		private static char[] str = new char[500*8*2];
		private static boolean[] isDigit = new boolean[256];
		private static boolean[] isSpace = new boolean[256];
		private static boolean[] isLineSep = new boolean[256];

		static { for(int i = 0; i < 10; i++) { isDigit['0' + i] = true; } isDigit['-'] = true; isSpace[' '] = isSpace['\r'] = isSpace['\n'] = isSpace['\t'] = true; isLineSep['\r'] = isLineSep['\n'] = true; }
		public int read() throws IOException { if(pos >= readLen) { pos = 0; readLen = in.read(buffer); if(readLen <= 0) { throw new EndOfFileRuntimeException(); } } return buffer[pos++]; }
		public int nextInt() throws IOException { int len = 0; str[len++] = nextChar(); len = reads(len, isSpace); int i = 0; int ret = 0; if(str[0] == '-') { i = 1; } for(; i < len; i++) ret = ret * 10 + str[i] - '0'; if(str[0] == '-') { ret = -ret; } return ret; }
		public long nextLong() throws IOException { int len = 0; str[len++] = nextChar(); len = reads(len, isSpace); int i = 0; long ret = 0; if(str[0] == '-') { i = 1; } for(; i < len; i++) ret = ret * 10 + str[i] - '0'; if(str[0] == '-') { ret = -ret; } return ret; }
		public char nextChar() throws IOException { while(true) { final int c = read(); if(!isSpace[c]) { return (char)c; } } }
		int reads(int len, boolean[] accept) throws IOException { try { while(true) { final int c = read(); if(accept[c]) { break; } if(str.length == len) { char[] rep = new char[str.length * 3 / 2]; System.arraycopy(str, 0, rep, 0, str.length); str = rep; } str[len++] = (char)c; } } catch(EndOfFileRuntimeException e) { ; } return len; }
		int reads(char[] cs, int len, boolean[] accept) throws IOException { try { while(true) { final int c = read(); if(accept[c]) { break; } cs[len++] = (char)c; } } catch(EndOfFileRuntimeException e) { ; } return len; }
		public char[] nextLine() throws IOException { int len = 0; str[len++] = nextChar(); len = reads(len, isLineSep); try { if(str[len-1] == '\r') { len--; read(); } } catch(EndOfFileRuntimeException e) { ; } return Arrays.copyOf(str, len); }
		public String nextString() throws IOException { return new String(next()); }
		public char[] next() throws IOException { int len = 0; str[len++] = nextChar(); len = reads(len, isSpace); return Arrays.copyOf(str, len); }
//		public int next(char[] cs) throws IOException { int len = 0; cs[len++] = nextChar(); len = reads(cs, len, isSpace); return len; }
		public double nextDouble() throws IOException { return Double.parseDouble(nextString()); }
		public long[] nextLongArray(final int n) throws IOException { final long[] res = new long[n]; for(int i = 0; i < n; i++) { res[i] = nextLong(); } return res; }
		public int[] nextIntArray(final int n) throws IOException { final int[] res = new int[n]; for(int i = 0; i < n; i++) { res[i] = nextInt(); } return res; }
		public int[][] nextIntArray2D(final int n, final int k) throws IOException { final int[][] res = new int[n][]; for(int i = 0; i < n; i++) { res[i] = nextIntArray(k); } return res; }
		public int[][] nextIntArray2DWithIndex(final int n, final int k) throws IOException { final int[][] res = new int[n][k+1]; for(int i = 0; i < n; i++) { for(int j = 0; j < k; j++) { res[i][j] = nextInt(); } res[i][k] = i; } return res; }
		public double[] nextDoubleArray(final int n) throws IOException { final double[] res = new double[n]; for(int i = 0; i < n; i++) { res[i] = nextDouble(); } return res; }
	}
}
0