067 - Add Binary

最新推荐文章于 2022-06-30 09:39:46 发布

shuimuyq

最新推荐文章于 2022-06-30 09:39:46 发布

阅读量290

点赞数

分类专栏： leetcode

本文链接：https://blog.csdn.net/shuimuyq/article/details/50435667

版权

leetcode 专栏收录该内容

70 篇文章 0 订阅

订阅专栏

Given two binary strings, return their sum (also a binary string).

For example,
a = "11"
b = "1"
Return "100".

char* addBinary(char* a, char* b) 
{
	int lena = strlen(a);
	int lenb = strlen(b);
	int retlen = lena > lenb ? lena + 2 : lenb + 2;
	char *ret = (char *)malloc(sizeof(char) * retlen);
	ret[0] = '0';
	ret[--retlen] = '\0';
	int mark = 0;

	char *aend = a + lena - 1;
	char *bend = b + lenb - 1;
	char *rend = ret + retlen - 1;
	while (rend - ret > 0) {
		if (aend && bend)
			*rend = *aend + *bend + mark - '0';
		else if (aend)
			*rend = *aend + mark;
		else if (bend)
			*rend = *bend + mark;
		if (*rend >= '2') {
			*rend -= 2;
			mark = 1;
		} else
			mark = 0;
		if (aend && aend == a)
			aend = NULL;
		if (bend && bend == b)
			bend = NULL;
		if (aend) aend--;
		if (bend) bend--;
		rend--;
	}
	if (mark)
		ret[0] = '1';
	return ret[0] == '0' ? ret + 1 : ret;
}