글 읽기 - 반례들을 여럿 대입해봐도 제대로 작동하는 것 같은데 틀렸다고만 나옵니다

juyoung35 2년 전 0

반례들을 여럿 대입해봐도 제대로 작동하는 것 같은데 틀렸다고만 나옵니다

use std::io::{stdin, prelude::*};
fn digits(n: usize, k: usize, var: &mut usize) -> Option<Vec<u8>> {
    for i in 0..=9 {
        *var += (pow(k) + 1) * i;
        if k > 1 {
            match digits(n, k - 1, var) {
                 // 3. 벡터를 쌓아갑니다.
                Some(v) => return Some(vec![i as u8].into_iter().chain(v.into_iter()).collect()),
                 // 4. 아닐 경우에는 더한 값을 다시 빼줍니다.
                None => *var -= (pow(k) + 1) * i,
            }
        } else { // 1. k == 1이면 일의 자리 수라는 것인데 이 때의 쌓아올린 var 값이 n과 같으면 Some()을 반환하며
                 // 2. 재귀적으로 건너온 과정을 역으로 거슬러가며
            if *var == n {
                return Some(vec![i as u8]);
            } else {
                 // 4. 아닐 경우에는 더한 값을 다시 빼줍니다.
                *var -= (pow(k) + 1) * i;
            }
        }
    }
    None
}
// k 번째 자릿수를 더해주기 위한 함수
fn pow(k: usize) -> usize {
    if k <= 1 { 1 } else { 10 * pow(k - 1) }
}
fn main() {
    let v: Vec<u8> = stdin().bytes().map(|b| b.unwrap() - b'0').collect();
    let mut n = 0;
    for i in 0..v.len() {
        n += v[i] as usize * pow(v.len() - i);
    }
    let f = v[0];
    'l: loop {
        for i in 0..=f { // 첫 번째 자리 수에 대해서만 개별 루프를 돌려줍니다.
            let mut var = (pow(v.len()) + 1) * i as usize;
            match digits(n, v.len() - 1, &mut var) {
                Some(v) => {
                    if i > 0 { print!("{}", i) } // 첫 번째 자리 수가 0이면 출력하지 않습니다.
                    v.iter().for_each(|x| print!("{}", x));
                    break 'l; // 'l 루프를 빠져나가므로
                },            //
                None => (),   //
            }                 //
        }                     //
        print!("0");          // 이 출력이 무시됩니다.
        break;
    }
}

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

반례들을 여럿 대입해봐도 제대로 작동하는 것 같은데 틀렸다고만 나옵니다

juyoung35 2년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말