HDU 2024 C语言合法标识符

http://acm.hdu.edu.cn/showproblem.php?pid=2024

Problem Description

输入一个字符串，判断其是否是C的合法标识符。

 

Input

输入数据包含多个测试实例，数据的第一行是一个整数n,表示测试实例的个数，然后是n行输入数据，每行是一个长度不超过50的字符串。

 

Output

对于每组输入数据，输出一行。如果输入数据是C的合法标识符，则输出"yes"，否则，输出“no”。

 

Sample Input

3

12ajf

fi8x_a

ff ai_2

 

Sample Output

no

yes

no

 

代码：

#include <bits/stdc++.h>

using namespace std;

char s[111];
int len;

int A()
{
    if(s[0]>='a'&&s[0]<='z'||s[0]>='A'&&s[0]<='Z'||s[0]=='_')
        return 1;
    else
        return 0;
}
int B()
{
    for(int i=1; i<len; i++)
    {
        if(s[i]>='a'&&s[i]<='z'||s[i]>='A'&&s[i]<='Z'||s[i]=='_'||
           s[i]>='0'&&s[i]<='9')
            continue;
        else
            return 0;
    }
    return 1;
}

int main()
{
    int n;
    cin>>n;
    getchar();
    int sum=1;
    for(int i=1; i<=n; i++)
    {
        gets(s);
        //printf("!!!%s\n", s);
        len = strlen(s);
        sum=A()*B();
        //cout<<A()<<" "<<B()<<endl;
        if(sum==1&&len<=50)
            cout<<"yes"<<endl;
        else
            cout<<"no"<<endl;
    }
    return 0;
}