The Same Color

Description

Diao Yang has many balls with different colors. Today he wants to divide his balls into two groups. But he does not know how to divide. Then Diao Ze gives him a suggestion: first you choose a ball and put it into the first group. Then from the second ball, if the color of the ball is same to the last ball you has chosen, then you put the ball into the other group. Otherwise you put the ball into the same group with the last ball. Diao Yang thinks it is OK.
Then the problem is, Diao Yang wants to know the product of the number of two groups’ balls. Can you help him?
Input

The first line contains an integer T, indicating the number of test cases.
In each test case, there are several lines, each line contains only one string with lowercas, indicating the color of the ball Diao Yang has chosen currently. Diao Yang will stop choosing when the string is equal to “END”. Note that the string “END” does not mean a color.
You can assume that there are at most 100 lines in each test case and each string has at most 10 letters.
Output

For each test case, output the answer in a line.
Sample Input
3
yellow
yellow
pink
red
red
green
END
blue
black
purple
purple
END
rose
orange
brown
white
END
Sample Output
9
3
0
HINT

In the first test case, the color of the first group’s balls are
yellow, red, green, 3 balls in total. The color of the second group’s
balls are yellow, pink, red, 3 balls in total too. So the product is
3×3=9.

In the second test case, the answer is 3×1=3 and in the third test
case the answer is 4×0=0.

#include<iostream>
using namespace std;
#include<stdio.h>
int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        int s1=0;
        int s2=0,flag=0;
        string per="";
        string xz;
        int r;
        for(int i=0;i<99999;i++)
        {
             cin>>xz;
            if(xz.compare("END")==0)
             {
                     printf("%d\n",s1*s2);
                     break;
             }
             if(xz.compare(per)==0)
             {

                 ++s2;
                 flag++;
                 if(flag>0)
                 {
                     r=s1;
                     s1=s2;
                     s2=r;
                 }
             }
             else
                ++s1;
             per=xz;
        }
    }
    return 0;
}