竞选社长/计票-c++

题目描述
假设你们社团要竞选社长，有两名候选人分别是A和B，社团每名同学必须并且只能投一票，最终得票多的人为社长.
输入描述:
一行，字符序列，包含A或B，输入以字符0结束。
输出描述:
一行，一个字符，A或B或E，输出A表示A得票数多，输出B表示B得票数多，输出E表示二人得票数相等。
示例1
输入
ABBABBAAB0
输出
B

#include<iostream>
using namespace std;
int main ()
{
    int a_count=0,b_count=0;
    char ch;
    while(cin>>ch)
    {
        if(ch==0)
            break;
        if(ch=='A')
            a_count++;
        if(ch=='B')
            b_count++;
    }
    if(a_count==b_count)
        cout<<"E"<<endl;
    else if(a_count>b_count)
        cout<<"A"<<endl;
    else
        cout<<"B"<<endl;
    return 0;
}