Submission #958343


Source Code Expand

import java.io.IOException;
import java.io.InputStream;
import java.util.*;
import java.util.function.BiFunction;
import java.util.function.Function;
import java.util.function.Supplier;

public class Main {
  void run() {
    int n = ni();
    long[] x = new long[n + 1];
    for (int i = 1; i <= n; ++i) {
      x[i] = ni();
    }
    int m = ni();
    long k = sc.nextLong();
    int[] a = new int[m];
    for (int i = 0; i < m; ++i) {
      a[i] = ni();
    }

    long[] d = new long[n];
    for (int i = 1; i <= n - 1; ++i) {
      d[i] = x[i + 1] - x[i];
    }
    int[] tikanD = new int[n + 1];
    for (int i = 1; i <= n; ++i) {
      tikanD[i] = i;
    }
    for (int i = 0; i < m; ++i) {
      int left = a[i] - 1;
      int right = a[i];
      int tmp = tikanD[left];
      tikanD[left] = tikanD[right];
      tikanD[right] = tmp;
    }
    tikanD = pow_tikan(tikanD, k);
    long[] d_ = new long[n];
    for (int i = 1; i <= n - 1; ++i) {
      d_[i] = d[tikanD[i]];
    }
    d = d_;
    long ite = x[1];
    System.out.println(ite);
    for (int i = 1; i <= n - 1; ++i) {
      ite += d[i];
      System.out.println(ite);
    }
  }

  int[] tikan(int[] a, int[] b) {
    int[] res = new int[a.length];
    for (int i = 0; i < a.length; ++i) {
      res[i] = i;
    }
    for (int i = 0; i < a.length; ++i) {
      res[i] = b[a[res[i]]];
    }
    return res;
  }

  int[] pow_tikan(int[] d, long k) {
    int[] res = new int[d.length];
    for (int i = 0; i < d.length; ++i) {
      res[i] = i;
    }
    int[] v = d.clone();
    while (k > 0) {
      if (k % 2 == 1) {
        res = tikan(res, v);
      }
      v = tikan(v, v);
      k /= 2;
    }
    return res;
  }

  Scanner sc = new Scanner(System.in);

  public static void main(String[] args) {
    new Main().run();
  }

  int ni() {
    return Integer.parseInt(sc.next());
  }

  void debug(Object... os) {
    System.err.println(Arrays.deepToString(os));
  }

  class BIT<T> {
    int n;
    ArrayList<T> bit;
    BiFunction<T, T, T> bif;

    /**
     * 1-indexed なBinary Indexed Treeを構築する
     *
     * @param n   容量
     * @param bif 適用させる関数
     * @param sup 初期値
     */
    BIT(int n, BiFunction<T, T, T> bif, Supplier<T> sup) {
      this.n = n;
      bit = new ArrayList<>(n + 1);
      for (int i = 0; i < n + 1; ++i) {
        bit.add(sup.get());
      }
      this.bif = bif;
    }

    /**
     * iの位置の値をvで更新する
     *
     * @param i index
     * @param v 新しい値
     */
    void update(int i, T v) {
      for (int x = i; x <= n; x += x & -x) {
        bit.set(x, bif.apply(bit.get(x), v));
      }
    }

    /**
     * クエリー
     *
     * @param defaultValue 初期値
     * @param i            index
     * @return [1, i]までfを適用した結果
     */
    T reduce(T defaultValue, int i) {
      T ret = defaultValue;
      for (int x = i; x > 0; x -= x & -x) {
        ret = bif.apply(ret, bit.get(x));
      }
      return ret;
    }
  }

  long MOD = 1_000_000_007;

  /**
   * 繰り返し2乗法を用いたべき乗の実装
   *
   * @return a^r (mod 1,000,000,007)
   */
  long pow(long a, long r) {
    long sum = 1;
    while (r > 0) {
      if ((r & 1) == 1) {
        sum *= a;
        sum %= MOD;
      }
      a *= a;
      a %= MOD;
      r >>= 1;
    }
    return sum;
  }

  /**
   * 組み合わせ
   * O(n)
   *
   * @return {}_nC_r
   */
  long C(int n, int r) {
    long sum = 1;
    for (int i = n; 0 < i; --i) {
      sum *= i;
      sum %= MOD;
    }
    long s = 1;
    for (int i = r; 0 < i; --i) {
      s *= i;
      s %= MOD;
    }
    sum *= pow(s, MOD - 2);
    sum %= MOD;

    long t = 1;
    for (int i = n - r; 0 < i; --i) {
      t *= i;
      t %= MOD;
    }
    sum *= pow(t, MOD - 2);
    sum %= MOD;

    return sum;
  }

  double GOLDEN_RATIO = (1.0 + Math.sqrt(5)) / 2.0;

  /**
   * 黄金分割探索
   *
   * @param left  下限
   * @param right 上限
   * @param f     探索する関数
   * @param comp  上に凸な関数を探索するときは、Comparator.comparingDouble(Double::doubleValue)
   *              下に凸な関数を探索するときは、Comparator.comparingDouble(Double::doubleValue).reversed()
   * @return 極値の座標x
   */
  double goldenSectionSearch(double left, double right, Function<Double, Double> f, Comparator<Double> comp) {
    double c1 = divideInternally(left, right, 1, GOLDEN_RATIO);
    double c2 = divideInternally(left, right, GOLDEN_RATIO, 1);
    double d1 = f.apply(c1);
    double d2 = f.apply(c2);
    while (right - left > 1e-9) {
      if (comp.compare(d1, d2) > 0) {
        right = c2;
        c2 = c1;
        d2 = d1;
        c1 = divideInternally(left, right, 1, GOLDEN_RATIO);
        d1 = f.apply(c1);
      } else {
        left = c1;
        c1 = c2;
        d1 = d2;
        c2 = divideInternally(left, right, GOLDEN_RATIO, 1);
        d2 = f.apply(c2);
      }
    }
    return right;
  }

  /**
   * [a,b]をm:nに内分する点を返す
   */
  double divideInternally(double a, double b, double m, double n) {
    return (n * a + m * b) / (m + n);
  }

  /**
   * http://qiita.com/p_shiki37/items/65c18f88f4d24b2c528b
   */
  static class FastScanner {
    private final InputStream in;
    private final byte[] buffer = new byte[1024];
    private int ptr = 0;
    private int buflen = 0;

    public FastScanner(InputStream in) {
      this.in = in;
    }

    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 static 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 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();
      }
    }
  }
}

Submission Info

Submission Time
Task C - Rabbit Exercise
User arukuka
Language Java8 (OpenJDK 1.8.0)
Score 800
Code Size 7415 Byte
Status AC
Exec Time 1810 ms
Memory 57600 KB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 800 / 800
Status
AC × 3
AC × 29
Set Name Test Cases
Sample 0_00.txt, 0_01.txt, 0_02.txt
All 0_00.txt, 0_01.txt, 0_02.txt, 1_00.txt, 1_01.txt, 1_02.txt, 1_03.txt, 1_04.txt, 1_05.txt, 1_06.txt, 1_07.txt, 1_08.txt, 1_09.txt, 1_10.txt, 1_11.txt, 1_12.txt, 1_13.txt, 1_14.txt, 1_15.txt, 1_16.txt, 1_17.txt, 1_18.txt, 1_19.txt, 1_20.txt, 1_21.txt, 1_22.txt, 1_23.txt, 1_24.txt, 1_25.txt
Case Name Status Exec Time Memory
0_00.txt AC 125 ms 9800 KB
0_01.txt AC 126 ms 9664 KB
0_02.txt AC 128 ms 9672 KB
1_00.txt AC 128 ms 9552 KB
1_01.txt AC 128 ms 9676 KB
1_02.txt AC 1565 ms 46248 KB
1_03.txt AC 1647 ms 50736 KB
1_04.txt AC 1375 ms 50872 KB
1_05.txt AC 286 ms 21996 KB
1_06.txt AC 339 ms 29812 KB
1_07.txt AC 349 ms 28584 KB
1_08.txt AC 374 ms 31112 KB
1_09.txt AC 344 ms 30440 KB
1_10.txt AC 252 ms 20956 KB
1_11.txt AC 329 ms 28220 KB
1_12.txt AC 175 ms 12476 KB
1_13.txt AC 380 ms 30940 KB
1_14.txt AC 358 ms 30012 KB
1_15.txt AC 498 ms 32052 KB
1_16.txt AC 349 ms 19804 KB
1_17.txt AC 1152 ms 43784 KB
1_18.txt AC 1155 ms 34732 KB
1_19.txt AC 1404 ms 57600 KB
1_20.txt AC 839 ms 40132 KB
1_21.txt AC 846 ms 32352 KB
1_22.txt AC 1415 ms 54336 KB
1_23.txt AC 1437 ms 43316 KB
1_24.txt AC 1211 ms 34288 KB
1_25.txt AC 1810 ms 57344 KB