結果

問題 No.1318 ABCD quadruplets
コンテスト
ユーザー 箱星
提出日時 2020-12-21 12:23:05
言語 Kotlin
(2.1.0)
結果
TLE  
実行時間 -
コード長 1,653 bytes
コンパイル時間 13,469 ms
コンパイル使用メモリ 444,796 KB
実行使用メモリ 95,148 KB
最終ジャッジ日時 2024-09-21 12:54:02
合計ジャッジ時間 39,914 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 26 TLE * 4
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

fun PrintWriter.solve() {
    val n = nextInt()
    val m = nextInt()
    val ans = Array(n + 1) { 0 }
    for (a in 0..m) {
        for (b in 0..a) {
            for (c in 0..b) {
                for (d in 0..c) {
                    val k = a * a + a * b + a * c + a * d + b * b + b * c + b * d + c * c + c * d + d * d
                    if (k > n) break
                    if (a == b) {
                        if (b == c) {
                            if (c == d) ans[k] += 1
                            else ans[k] += 4
                        } else {
                            if (c == d) ans[k] += 6
                            else ans[k] += 12
                        }
                    } else {
                        if (b == c) {
                            if (c == d) ans[k] += 4
                            else ans[k] += 12
                        } else {
                            if (c == d) ans[k] += 12
                            else ans[k] += 24
                        }
                    }
                }
            }
        }
    }
    println(ans.joinToString("\n"))
}

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
0