結果
問題 | No.1084 積の積 |
ユーザー |
![]() |
提出日時 | 2020-06-19 22:57:14 |
言語 | Java (openjdk 23) |
結果 |
AC
|
実行時間 | 214 ms / 2,000 ms |
コード長 | 9,461 bytes |
コンパイル時間 | 3,003 ms |
コンパイル使用メモリ | 79,780 KB |
実行使用メモリ | 56,200 KB |
最終ジャッジ日時 | 2024-07-03 15:24:21 |
合計ジャッジ時間 | 8,718 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 5 |
other | AC * 27 |
ソースコード
import java.io.*;import java.math.BigDecimal;import java.util.*;@SuppressWarnings("unused")public class Main {FastScanner in;PrintWriter out;final static int MOD = (int)1e9+7;void solve(){int N = in.nextInt();long[] A = in.nextLongArray(N);for(int i = 0; i < N; i++){if(A[i] == 0){out.println("0");return;}}long[] mul = new long[N];mul[0] = A[0];for(int i = 1; i < N; i++){mul[i] = mul[i-1] * A[i];}DualSegmentTree tree = new DualSegmentTree(N);int idxR = 1;long m = A[0];for(int i = 0; i < N; i++){while(idxR < N && m * A[idxR] < (int)1e9){m *= A[idxR];idxR++;}tree.setSegment(i, idxR, 1);if(i > 0) tree.setSegment(i - 1, i, -(idxR - i));m /= A[i];}long sum = 0, ans = 1;for(int i = N - 1; i >= 0; i--){sum += tree.getPoint(i);//out.println(sum);ans *= modPow(A[i], sum, MOD);ans %= MOD;}out.println(ans);}public static long modPow(long a, long b, int mod){long res = 1;while(b > 0){if((b & 1) == 1) res = (res * a) % mod;a = (a * a) % mod;b = b >> 1;}return res;}class DualSegmentTree{int d;long[] node;/*作用素で使える単位元*/private long e = 0;/*結合律が成り立ち、更新が可換であるような、各要素への作用素*/private long f(long nodeVal, long val){return nodeVal + val;}/* 単位元で初期化 */public DualSegmentTree(int sz){long[] v = new long[sz];Arrays.fill(v, e);init(v, e);}/* 配列vで初期化 */public DualSegmentTree(long[] v){init(v, e);}private void init(long[] v, long e){this.e = e;d = 1;while(d < v.length) d *= 2;node = new long[2*d];Arrays.fill(node, e);for(int i = 0; i < v.length; i++)node[i+d] = v[i];}/* 0-indexed:xの要素を取得する */public long getPoint(int x){x += d;long res = node[x];while(x > 1){x = x / 2;res = f(res, node[x]);}return res;}/* 指定した区間[L,R)に含まれるすべての要素に作用素を適用するクエリ */public void setSegment(int L, int R, long val){L += d;R += d;while (L < R) {if ((L & 1) != 0){node[L] = f(node[L], val);L++;}if ((R & 1) != 0){--R;node[R] = f(node[R], val);}L >>= 1;R >>= 1;}}}public static void main(String[] args) {new Main().m();}private void m() {in = new FastScanner(System.in);out = new PrintWriter(System.out);/*try {String path = "input.txt";out = new PrintWriter(new BufferedWriter(new FileWriter(new File(path))));}catch (IOException e){e.printStackTrace();}*/solve();out.flush();in.close();out.close();}static class FastScanner {private Reader input;public FastScanner() {this(System.in);}public FastScanner(InputStream stream) {this.input = new BufferedReader(new InputStreamReader(stream));}public void close() {try {this.input.close();} catch (IOException e) {e.printStackTrace();}}public int nextInt() {long nl = nextLong();if (nl < Integer.MIN_VALUE || nl > Integer.MAX_VALUE) throw new NumberFormatException();return (int) nl;}public long nextLong() {try {int sign = 1;int b = input.read();while ((b < '0' || '9' < b) && b != '-' && b != '+') {b = input.read();}if (b == '-') {sign = -1;b = input.read();} else if (b == '+') {b = input.read();}long ret = b - '0';while (true) {b = input.read();if (b < '0' || '9' < b) return ret * sign;ret *= 10;ret += b - '0';}} catch (IOException e) {e.printStackTrace();return -1;}}public double nextDouble() {try {double sign = 1;int b = input.read();while ((b < '0' || '9' < b) && b != '-' && b != '+') {b = input.read();}if (b == '-') {sign = -1;b = input.read();} else if (b == '+') {b = input.read();}double ret = b - '0';while (true) {b = input.read();if (b < '0' || '9' < b) break;ret *= 10;ret += b - '0';}if (b != '.') return sign * ret;double div = 1;b = input.read();while ('0' <= b && b <= '9') {ret *= 10;ret += b - '0';div *= 10;b = input.read();}return sign * ret / div;} catch (IOException e) {e.printStackTrace();return Double.NaN;}}public char nextChar() {try {int b = input.read();while (Character.isWhitespace(b)) {b = input.read();}return (char) b;} catch (IOException e) {e.printStackTrace();return 0;}}public String nextStr() {try {StringBuilder sb = new StringBuilder();int b = input.read();while (Character.isWhitespace(b)) {b = input.read();}while (b != -1 && !Character.isWhitespace(b)) {sb.append((char) b);b = input.read();}return sb.toString();} catch (IOException e) {e.printStackTrace();return "";}}public String nextLine() {try {StringBuilder sb = new StringBuilder();int b = input.read();while (b != -1 && b != '\n') {sb.append((char) b);b = input.read();}return sb.toString();} catch (IOException e) {e.printStackTrace();return "";}}public int[] nextIntArray(int n) {int[] res = new int[n];for (int i = 0; i < n; i++) {res[i] = nextInt();}return res;}public int[] nextIntArrayDec(int n) {int[] res = new int[n];for (int i = 0; i < n; i++) {res[i] = nextInt() - 1;}return res;}public int[] nextIntArray1Index(int n) {int[] res = new int[n + 1];for (int i = 0; i < n; i++) {res[i + 1] = nextInt();}return res;}public long[] nextLongArray(int n) {long[] res = new long[n];for (int i = 0; i < n; i++) {res[i] = nextLong();}return res;}public long[] nextLongArrayDec(int n) {long[] res = new long[n];for (int i = 0; i < n; i++) {res[i] = nextLong() - 1;}return res;}public long[] nextLongArray1Index(int n) {long[] res = new long[n + 1];for (int i = 0; i < n; i++) {res[i + 1] = nextLong();}return res;}public double[] nextDoubleArray(int n) {double[] res = new double[n];for (int i = 0; i < n; i++) {res[i] = nextDouble();}return res;}}}