Kertoma

use std::io::stdin;

const ONE: [u64; 10] = [0, 1, 0, 0, 0, 0, 0, 0, 0, 0];
const TWO: [u64; 10] = [0, 0, 1, 0, 0, 0, 0, 0, 0, 0];
const THREE: [u64; 10] = [0, 0, 0, 0, 0, 0, 1, 0, 0, 0];
const FOUR: [u64; 10] = [0, 0, 1, 0, 1, 0, 0, 0, 0, 0];
const FIVE: [u64; 10] = [1, 1, 1, 0, 0, 0, 0, 0, 0, 0];
const SIX: [u64; 10] = [1, 0, 1, 0, 0, 0, 0, 1, 0, 0];


fn main() {
    let mut buf = String::new();
    stdin().read_line(&mut buf).unwrap();

    let mut digits = [0_u64; 10];

    let mut digits_sum: u64 = 0;
    for (i, digit) in buf.split_whitespace().enumerate() {
        let num = digit.parse::<u64>().unwrap();
        digits[i] += num;
        digits_sum += num;
    }

    if digits_sum < 4 {
        let i = match digits {
            ONE => 1,
            TWO => 2,
            THREE => 3,
            FOUR => 4,
            FIVE => 5,
            SIX => 6,
            _ => panic!(),
        };
        println!("{i}");
        return;
    }

    let mut sum: f64 = 0.0;
    for i in 1.. {
        sum += (i as f64).log10();

        if sum.floor() as u64 + 1 == digits_sum {
            println!("{i}");
            return;
        }
    }
}

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

Test 2

Group: 1, 2, 3

Verdict: ACCEPTED

Test 3

Group: 1, 2, 3

Verdict: ACCEPTED

Test 4

Group: 1, 2, 3

Verdict: ACCEPTED

Test 5

Group: 1, 2, 3

Verdict: ACCEPTED

Test 6

Group: 1, 2, 3

Verdict: ACCEPTED

Test 7

Group: 2, 3

Verdict: ACCEPTED

Test 8

Group: 2, 3

Verdict: ACCEPTED

Test 9

Group: 2, 3

Verdict: ACCEPTED

Test 10

Group: 2, 3

Verdict: ACCEPTED

Test 11

Group: 3

Verdict: ACCEPTED

Test 12

Group: 3

Verdict: ACCEPTED

Test 13

Group: 3

Verdict: ACCEPTED

Test 14

Group: 3

Verdict: ACCEPTED

Test 15

Group: 3

Verdict: ACCEPTED

Test 16

Group: 3

Verdict: ACCEPTED