[백준] 15740_A+B -9

[문제]: 15740번: A+B - 9 (acmicpc.net)

 

음수와 양수의 덧셈을 할 때는 절댓값이 작은 수에 9의 보수를 구해서 덧셈으로 바꾸어 풀었다.

두 수의 부호가 같다면 그냥 더하고 마지막에 부호를 결정만 한다.

 

#include <iostream>
#include <string>
#include <algorithm>
using namespace std;

string operator+(string a, string b)
{
	bool na = false;
	bool nb = false;
	string ret;

	if (a[0] == '-') {
		a.erase(begin(a));
		if (a[0] == '0')
			na = false;
		else
			na = true;
	}
	if (b[0] == '-') {
		b.erase(begin(b));
		if (b[0] == '0')
			na = false;
		else
			nb = true;
	}

	reverse(begin(a), end(a));
	reverse(begin(b), end(b));

	while (a.size() > b.size())
		b += '0';
	while (a.size() < b.size())
		a += '0';

	int altb = -1;	// a > b이면 1, a < b이면 0, a == b이면 -1
	for (int i = int(a.size()) - 1 ; i >= 0; --i) {
		if (a[i] > b[i]) {
			altb = 1;
			break;
		}
		else if (a[i] < b[i]) {
			altb = 0;
			break;
		}
	}

	if((na && !nb) || (!na && nb)) {	// 두 수중 하나가 음수라면
		if (altb == 1) {// 절댓값 b가 더 작다면
			for (int i = 0; i < b.size(); ++i)	// 각 자리의 9의 보수
				b[i] = '9' - b[i] + '0';
			reverse(begin(b), end(b));
			b = b + string("1");
			reverse(begin(b), end(b));

			if (b.size() > a.size())
				b.erase(end(b) - 1);

			int carry = 0;
			for (int i = 0; i < b.size(); ++i) {
				int s = (a[i] - '0') + (b[i] - '0');
				ret += (s + carry) % 10 + '0';
				carry = (s + carry) / 10;
			}
			
			reverse(begin(ret), end(ret));
			if (na)
				ret.insert(begin(ret), '-');
		}
		else if (altb == 0) {// 절댓값 a가 더 작다면
			for (int i = 0; i < a.size(); ++i)	// 각 자리의 9의 보수
				a[i] = '9' - a[i] + '0';
			reverse(begin(a), end(a));
			a = a + string("1");
			reverse(begin(a), end(a));

			if (a.size() > b.size())
				a.erase(end(a) - 1);

			int carry = 0;
			for (int i = 0; i < a.size(); ++i) {
				int s = (a[i] - '0') + (b[i] - '0');
				ret += (s + carry) % 10 + '0';
				carry = (s + carry) / 10;
			}

			reverse(begin(ret), end(ret));
			if (nb)
				ret.insert(begin(ret), '-');
		}
		else if (altb == -1) // 두 수의 절댓값이 같다면
			ret = "0";
	}
	else {	// 두 수의 부호가 같다면
		int carry = 0;
		for (int i = 0; i < a.size(); ++i) {
			int s = (a[i] - '0') + (b[i] - '0');
			ret += (s + carry) % 10 + '0';
			carry = (s + carry) / 10;
		}

		if (carry != 0)
			ret += carry + '0';

		reverse(begin(ret), end(ret));
		if ((na && nb) && altb != -1)
			ret.insert(begin(ret), '-');
	}
	
	return ret;
}

int main()
{
	ios_base::sync_with_stdio(false);
	cin.tie(nullptr);
	cout.tie(nullptr);
	string a, b;
	cin >> a >> b;
	string ret = a + b;
	char buho = ' ';
	for (int i = 0; i < ret.size(); ++i) {
		if (ret[i] == '-') {
			buho = '-';
			continue;
		}
		if (ret[i] != '0'){
			ret = ret.substr(i, ret.size() - i);
			break;
		}
	}
	if(buho == '-')
		ret.insert(begin(ret), buho);
	cout << ret << endl;
	return 0;
}

 

 

이게 브론즈 5 맞나..?

string 말고 BinInteger 클래스 만들어서도 풀어보자.