結果
問題 |
No.2046 Ans Mod? Mod Ans!
|
ユーザー |
|
提出日時 | 2021-07-18 21:51:24 |
言語 | Kotlin (2.1.0) |
結果 |
AC
|
実行時間 | 2,109 ms / 4,000 ms |
コード長 | 1,906 bytes |
コンパイル時間 | 14,575 ms |
コンパイル使用メモリ | 455,676 KB |
実行使用メモリ | 96,012 KB |
最終ジャッジ日時 | 2024-12-16 08:05:49 |
合計ジャッジ時間 | 33,383 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 18 |
ソースコード
import java.io.PrintWriter import java.util.* import kotlin.math.* // sum of A_j * (A_i / A_j) fun helper(a: Array<Int>): Long { val n = a.size val max = a.maxOrNull()!! val sqrt = sqrt(max.toDouble()).roundToInt() val l = Array(sqrt) { 0L } val bit = BIT(max + 1) var ans = 0L for (i in 0 until n) { if (a[i] < sqrt) { ans += l[a[i]] } else { for (k in 0..max step a[i]) { ans += k * bit.sum(k until minOf(k + a[i], max + 1)) } } bit.add(a[i], 1) for (j in 1 until sqrt) { l[j] += a[i] / j * j.toLong() } } return ans } fun PrintWriter.solve() { val n = nextInt() val a = Array(n) { nextInt() } a.sortDescending() var ans = helper(a) for (i in 0 until n) { ans += (2 * i - n + 1L) * a[i] } println(ans) } class BIT(private val n: Int) { private val data = LongArray(n) { 0 } fun add(i: Int, x: Long) { var p = i + 1 while (p <= n) { data[p - 1] += x p += p and -p } } fun sum(range: IntRange): Long { val l = range.first val r = range.last + 1 return sum(r) - sum(l) } private fun sum(i: Int): Long { var r = i var s = 0L while(r > 0) { s += data[r - 1] r -= r and -r } return s } } fun main() { val writer = PrintWriter(System.out, false) writer.solve() writer.flush() } // region Scanner private var st = StringTokenizer("") private val br = System.`in`.bufferedReader() fun next(): String { while (!st.hasMoreTokens()) st = StringTokenizer(br.readLine()) return st.nextToken() } fun nextInt() = next().toInt() fun nextLong() = next().toLong() fun nextLine() = br.readLine() fun nextDouble() = next().toDouble() // endregion