글 읽기 - 반례를 못 찾겠는데 50점이 나오네요 ㅠ 살려주세요

ecanus9 3년 전 0

Rust로 풀었는데, 웬만한 수는 다 집어 넣어봤는데 반례를 못 찾겠습니다 ㅠㅠ

﻿fn main() {
    let mut s = String::<i>new</i>();
    std::io::stdin().read_line(&mut s).unwrap();
    let mut iter = s.trim().split_whitespace();
    let mut a = iter.next().unwrap().as_bytes().to_vec();
    let mut b = iter.next().unwrap().as_bytes().to_vec();
    let mut a_is_neg = false;
    let mut b_is_neg = false;
    if a[0] == b'-' {a.remove(0); a_is_neg = true;}
    if b[0] == b'-' {b.remove(0); b_is_neg = true;}
    let mut result;
    if a[0] == b'0' && b[0] == b'0'{println!("0"); return}
    if a_is_neg && b_is_neg {
        result = add(&a,&b);
        result.insert(0, b'-');
    } else if a_is_neg || b_is_neg{
        result = sub(&a, &b);
        if a.len() == b.len(){
            for i in 0..a.len(){
                if a[i] != b[i]{
                    if a[i] < b[i]{
                        if b_is_neg {result.insert(0, b'-');}
                    } else{
                        if a_is_neg && result[0] != b'0' {result.insert(0, b'-');}
                    }
                    break;
                }
            }
        } else if a.len() < b.len(){
            if b_is_neg {result.insert(0, b'-');}
        } else{
            if a_is_neg {result.insert(0, b'-');}
        }
    } else{
        result = add(&a, &b);
    }
    let s:String = String::<i>from_utf8</i>(result).unwrap();
    println!("{}", s);

}

fn add(a:&Vec<u8>, b:&Vec<u8>) ->Vec<u8>{
    let mut big = if a.len() < b.len(){ b.clone()} else {a.clone()};
    let mut small = if a.len() < b.len(){ a.clone()} else {b.clone()};
    big.reverse();
    small.reverse();
    let mut result:Vec<u8> = big.clone();
    result.resize(big.len() + 1, b'0');
    let mut rem:u8 = 0;
    let mut res:u8;
    for i in 0..small.len(){
        res = rem + (big[i] - b'0') + (small[i] - b'0');
        rem = res / 10;
        result[i] = b'0' + res % 10;
    }
    let mut i = small.len();
    while rem != 0{
        result[i] += rem;
        rem = (result[i] - b'0') / 10;
        result[i] = b'0' + (result[i] - b'0') % 10;
        i += 1;
    }
    result.reverse();
    while result[0] == b'0'{
        result.remove(0);
    }
    result
}

fn sub(a:&Vec<u8>, b:&Vec<u8>) ->Vec<u8>{
    let mut big;
    let mut small;
    if a.len() == b.len(){
        if a.last() < b.last(){
            big = b.clone();
            small = a.clone();
        } else{
            big = a.clone();
            small = b.clone();
        }
    }
    else if a.len() < b.len(){
        big = b.clone();
        small = a.clone();
    }
    else {
        big = a.clone();
        small = b.clone();
    }
    big.reverse();
    small.reverse();
    let mut result:Vec<u8> = big.clone();
    result.resize(big.len() + 1, b'0');
    let mut rem:u8 = 0;
    let mut res:u8;
    for i in 0..small.len(){
        res = b'0' + (big[i] - b'0') - (small[i] - b'0') - rem;
        rem = if res < b'0'{
            res = res + 10;
            1
        } else{ 0 };
        result[i] = res;
    }
    let mut i = small.len();
    while rem != 0{
        result[i] -= rem;
        rem = if result[i] < b'0'{
            result[i] += 10;
            1
        } else { 0 };
        i += 1;
    }
    result.reverse();
    while result[0] == b'0' && result.len() != 1{
        result.remove(0);
    }
    if big[0] == b'-'{
        result.insert(0, b'-');
    }
    result
}

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

문제

문제

출처

ICPC

반례를 못 찾겠는데 50점이 나오네요 ㅠ 살려주세요

ecanus9 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말