結果

問題 No.2081 Make a Test Case of GCD Subset
ユーザー 箱星
提出日時 2022-09-25 22:10:22
言語 Kotlin
(2.1.0)
結果
AC  
実行時間 373 ms / 2,000 ms
コード長 1,919 bytes
コンパイル時間 16,744 ms
コンパイル使用メモリ 444,200 KB
実行使用メモリ 52,864 KB
最終ジャッジ日時 2024-12-22 14:41:17
合計ジャッジ時間 30,178 ms
ジャッジサーバーID
(参考情報)
judge4 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff #

import java.io.PrintWriter
import java.util.*
import kotlin.math.*

fun PrintWriter.solve() {
    val size = 100000
    val isPrime = BooleanArray(size) { true }
    val primes = mutableListOf<Int>()
    for (p in 2 until size) {
        if (!isPrime[p]) continue
        primes.add(p)
        var n = p * 2
        while (n < size) {
            isPrime[n] = false
            n += p
        }
    }
    var m = nextInt()
    if (m == 0) {
        println(1)
        println(1)
        return
    }
    fun helper(m: Int): Int {
        var v = 1
        var ret = 1
        while (2 * v + 1 <= m) {
            v = 2 * v + 1
            ret++
        }
        return ret
    }
    var m2 = m
    var ind = 0
    while (m2 > 0) {
        val r = helper(m2)
        m2 -= 1.shl(r) - 1
        ind++
    }
    var ind2 = ind - 1
    val ans = mutableListOf<Int>()
    while (m > 0) {
        val r = helper(m)
        m -= 1.shl(r) - 1
        for (i in 0 until r) {
            if (i == 0) {
                ans.add(primes[ind2])
            } else {
                ans.add(primes[ind] * primes[ind2])
                ind++
            }
        }
        ind2--
    }
    ans.sort()
    println(ans.size)
    println(ans.joinToString(" "))
}

fun main() {
    Thread(null, {
        val writer = PrintWriter(System.out, false)
        writer.solve()
        writer.flush()
    }, "solve", abs(1L.shl(26)))
        .apply { setUncaughtExceptionHandler { _, e -> e.printStackTrace(); kotlin.system.exitProcess(1) } }
        .apply { start() }.join()
}

// 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
0