Problem C: The Same Color 水题 字符串处理

Problem C: The Same Color
Time Limit: 1 Sec  Memory Limit: 128 MB
Submit: 993  Solved: 595
[Submit][Status][Web Board]
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.


来源： http://acm.hzau.edu.cn/problem.php?cid=1009&pid=2

#include <iostream>
#include <cstdio>
#include <map>
#include <string>
#include <algorithm>
#include <stdlib.h>
#include <cstring>
using namespace std;
char P[20];
string S,Ls;
#define a 1
#define b 2
int Handle(void)
{
    int A=0,B=0,flag = a;
    scanf("%s",P);Ls = P;A++;
    while(scanf("%s",P))
    {
        S = P;
        if(!S.compare("END")) break;
        else
        {
            if(Ls.compare(S)){if(flag == a) A++;else B++;}
            else
            {if(flag == a) {B++;flag=b;}else {A++;flag=a;}}
            Ls = S;
        }
    }
   // printf("A==%d B==%d\n",A,B);
    return A*B;
}
int main()
{
    //freopen("D:\\test.txt","r",stdin);
    //freopen("D:\\tzzt.out","w",stdout);
    int T;cin>>T;
    for(int i=1;i<=T;i++) printf("%d\n",Handle());

    return 0;
}