結果

問題 No.1097 Remainder Operation
ユーザー ywmtywmt
提出日時 2020-06-26 22:10:34
言語 Java19
(openjdk 21)
結果
AC  
実行時間 1,418 ms / 2,000 ms
コード長 4,056 bytes
コンパイル時間 2,993 ms
コンパイル使用メモリ 74,780 KB
実行使用メモリ 79,076 KB
最終ジャッジ日時 2023-09-18 04:57:10
合計ジャッジ時間 20,334 ms
ジャッジサーバーID
(参考情報)
judge13 / judge11
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 42 ms
49,208 KB
testcase_01 AC 44 ms
49,128 KB
testcase_02 AC 56 ms
49,280 KB
testcase_03 AC 56 ms
49,424 KB
testcase_04 AC 56 ms
49,356 KB
testcase_05 AC 57 ms
49,232 KB
testcase_06 AC 57 ms
49,720 KB
testcase_07 AC 323 ms
55,704 KB
testcase_08 AC 330 ms
56,112 KB
testcase_09 AC 330 ms
56,028 KB
testcase_10 AC 322 ms
55,852 KB
testcase_11 AC 309 ms
55,752 KB
testcase_12 AC 1,072 ms
59,764 KB
testcase_13 AC 1,059 ms
59,764 KB
testcase_14 AC 1,325 ms
76,820 KB
testcase_15 AC 1,040 ms
58,988 KB
testcase_16 AC 1,086 ms
59,948 KB
testcase_17 AC 1,088 ms
59,916 KB
testcase_18 AC 1,336 ms
76,972 KB
testcase_19 AC 1,173 ms
58,256 KB
testcase_20 AC 1,368 ms
79,076 KB
testcase_21 AC 1,345 ms
76,608 KB
testcase_22 AC 1,418 ms
77,400 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

import java.io.IOException;
import java.io.InputStream;
import java.io.PrintWriter;
import java.util.Arrays;
import java.util.InputMismatchException;
import java.util.NoSuchElementException;

public class Main {
	static PrintWriter out;
	static InputReader ir;

	static void solve() {
		int n = ir.nextInt();
		long[] a = ir.nextLongArray(n);
		int q = ir.nextInt();
		long[] b = new long[n];
		for (int i = 0; i < n; i++) {
			b[i] = (a[i] + i) % n;
		}
		int[] f = new int[n];
		Arrays.fill(f, -1);
		int cur = 0;
		f[cur] = 0;
		int l = -1;
		long t = -1;
		long[] sum = new long[n];
		sum[cur] = a[0];
		while (true) {
			if (f[(int) b[cur]] >= 0) {
				t = f[cur] - f[(int) b[cur]] + 1;
				l = (int) b[cur];
				break;
			}
			f[(int) b[cur]] = f[cur] + 1;
			sum[(int) f[(int) b[cur]]] = sum[(int) f[cur]] + a[(int) b[cur]];
			cur = (int) b[cur];
		}
		tr(t, l);
		while (q-- > 0) {
			long k = ir.nextLong() - 1;
			if (k <= f[l]) {
				out.println(sum[(int) k]);
			} else {
				long res = sum[(int) f[l]];
				tr(res);
				k -= f[l];
				res += (sum[(int) (f[l] + t - 1)] - (f[l] == 0 ? 0 : sum[(int) f[l] - 1])) * (k / t);
				tr(sum[(int) (f[l] + t - 1)] - (f[l] == 0 ? 0 : sum[(int) f[l] - 1]));
				k %= t;
				res += sum[f[l] + (int) k] - sum[f[l]];
				out.println(res);
			}
		}
	}

	public static void main(String[] args) {
		ir = new InputReader(System.in);
		out = new PrintWriter(System.out);
		solve();
		out.flush();
	}

	static class InputReader {

		private InputStream in;
		private byte[] buffer = new byte[1024];
		private int curbuf;
		private int lenbuf;

		public InputReader(InputStream in) {
			this.in = in;
			this.curbuf = this.lenbuf = 0;
		}

		public boolean hasNextByte() {
			if (curbuf >= lenbuf) {
				curbuf = 0;
				try {
					lenbuf = in.read(buffer);
				} catch (IOException e) {
					throw new InputMismatchException();
				}
				if (lenbuf <= 0)
					return false;
			}
			return true;
		}

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

		private boolean isSpaceChar(int c) {
			return !(c >= 33 && c <= 126);
		}

		private void skip() {
			while (hasNextByte() && isSpaceChar(buffer[curbuf]))
				curbuf++;
		}

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

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

		public int nextInt() {
			if (!hasNext())
				throw new NoSuchElementException();
			int c = readByte();
			while (isSpaceChar(c))
				c = readByte();
			boolean minus = false;
			if (c == '-') {
				minus = true;
				c = readByte();
			}
			int res = 0;
			do {
				if (c < '0' || c > '9')
					throw new InputMismatchException();
				res = res * 10 + c - '0';
				c = readByte();
			} while (!isSpaceChar(c));
			return (minus) ? -res : res;
		}

		public long nextLong() {
			if (!hasNext())
				throw new NoSuchElementException();
			int c = readByte();
			while (isSpaceChar(c))
				c = readByte();
			boolean minus = false;
			if (c == '-') {
				minus = true;
				c = readByte();
			}
			long res = 0;
			do {
				if (c < '0' || c > '9')
					throw new InputMismatchException();
				res = res * 10 + c - '0';
				c = readByte();
			} while (!isSpaceChar(c));
			return (minus) ? -res : res;
		}

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

		public int[] nextIntArray(int n) {
			int[] a = new int[n];
			for (int i = 0; i < n; i++)
				a[i] = nextInt();
			return a;
		}

		public long[] nextLongArray(int n) {
			long[] a = new long[n];
			for (int i = 0; i < n; i++)
				a[i] = nextLong();
			return a;
		}

		public char[][] nextCharMap(int n, int m) {
			char[][] map = new char[n][m];
			for (int i = 0; i < n; i++)
				map[i] = next().toCharArray();
			return map;
		}
	}

	static void tr(Object... o) {
		System.err.println(Arrays.deepToString(o));
	}
}
0