ZOJ 2478 Encoding

#include "iostream"
#include "string"
using namespace std;

int main()
{
int TestCase;
cin >> TestCase;
while (TestCase--)
{
string str, first;
int length, count = 0;
cin >> str;
length = str.size();
first = str[0];
for (int i = 0; i < length; i++)
{
if (str[i] == first[0])
{
count++;
if (i == length - 1)
{
if (count == 1)
cout << first;
else
cout << count << first;
}
}
else
{
if (count == 1)
cout << first;
else
cout << count << first;
first = str[i];
count = 1;
if (i == length - 1)
{
if (count == 1)
cout << str[i];
else
cout << count << str[i];
}
}
}
cout << endl;
}
}