글 읽기 - 10757 질문

natwo2005 2년 전 0

틀렸습니다 라고 뜨는데 뭐가 잘못된걸까요?..

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

int main() {
    string a,b;
    cin >> a >> b;
    
    int la=a.length();
    int lb=b.length();
    
    vector<int> v;
    
    if(la==lb){
        int flag=0;
        int l=la;
        for(int i=l-1; i>=0; i--){
            int ai=int(a[i])-'0';
            int bi=int(b[i])-'0';
            int sum = ai+bi;
            if(flag==1){
                sum+=1;
                flag=0;
            }
            if(sum>=10){
                flag = 1;
                if(i==0){
                    flag=3;
                }
                sum-=10;
            }
            v.push_back(sum);
        }
        if(flag==3){
            v.push_back(1);
        }
    }    
    
    if(la>lb){
        string temp = a;
        a = b;
        b = temp;
        la=a.length();
        lb=b.length();
    
        int flag=0;
        
        for(int i=lb-1; i>=lb-la; i--){
            int ai=int(a[i-lb+la])-'0';
            int bi=int(b[i])-'0';
            int sum = ai+bi;
            if(flag==1){
                sum+=1;
                flag=0;
            }
            if(sum>=10){
                flag = 1;
                sum-=10;
            }
            v.push_back(sum);
        }
        for(int i=lb-la-1; i>=0; i--){
            int sum2=b[i]-'0';
            if(flag==1){
                sum2+=1;
            }
            v.push_back(sum2);
        }
    }
    
    reverse(v.begin(),v.end());
    vector<int>::iterator iter;
    
    for(iter=v.begin(); iter!=v.end(); iter++){
        cout << *iter;
    }
   
    return 0;
}

dinky24 2년 전 1

현재 풀이에서는 la < lb 의 경우도 고려해야할듯 합니다.
좀 더 일반화하면 la, lb에 관계없이 모든 경우에 사용할 수 있는 알고리즘을 계획할 수 있습니다.

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

문제

문제

출처

ICPC

10757 질문

natwo2005 2년 전 0

dinky24 2년 전 1

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말