PAT B1057 数零壹 (20分)

题目链接：https://pintia.cn/problem-sets/994805260223102976/problems/994805270914383872

题目描述
给定一串长度不超过 10^​5​​ 的字符串，本题要求你将其中所有英文字母的序号（字母 a-z 对应序号 1-26，不分大小写）相加，得到整数 N，然后再分析一下 N 的二进制表示中有多少 0、多少 1。例如给定字符串 PAT (Basic)，其字母序号之和为：16+1+20+2+1+19+9+3=71，而 71 的二进制是 1000111，即有 3 个 0、4 个 1。

输入
输入在一行中给出长度不超过 10^​5​​ 、以回车结束的字符串。

输出
在一行中先后输出 0 的个数和 1 的个数，其间以空格分隔。

样例输入
PAT (Basic)

样例输出
3 4

代码

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

int main() {
	string str;
	int zero_num = 0, one_num = 0, sum = 0;
	getline(cin, str);
	for(int i = 0; i < str.size(); i++) {
		char c = str[i];
		if(c >= 'A' && c <= 'Z')
			c = c - 'A' + 'a';
		if(c >= 'a' && c <= 'z') {
			sum += (c - 'a' + 1);
		}
	}
	while(sum > 0) {
		if(sum % 2 == 0)
			zero_num++;
		else
			one_num++;
		sum /= 2;
	}
	printf("%d %d\n", zero_num, one_num);
	return 0;
}