HDU2054 A==B? 情况考虑完全

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

Give you two numbers A and B, if A is equal to B, you should print "YES", or print "NO".

Input

each test case contains two numbers A and B. 

Output

for each case, if A is equal to B, you should print "YES", or print "NO".

Sample Input

1 2
2 2
3 3
4 3

Sample Output

NO
YES
YES
NO

#include<bits/stdc++.h>
using namespace std;
char a[1000000],b[1000000];
void init(char a[])
{
    int flag=0;
    for(int i=strlen(a)-1; i>=0; i--)
    {
        if(a[i]=='.')
        {
            flag=i;
            break;
        }
    }
    if(flag)
    {
        for(int i=strlen(a)-1; i>0; i--)
        {
            if(a[i]=='0')
            {
                a[i]='\0';
            }
            else
            {
                if(a[i]=='.'){
                    a[i]='\0';
                }
                break;
            }
        }
    }
}
int main()
{
    while(cin>>a>>b)
    {

        init(a);
        init(b);
        //cout<<a<<b<<endl;

        if(strlen(a)!=strlen(b)){
            cout<<"NO"<<endl;
            continue;
        }else{
            int flag=0;
            for(int i=0;i<strlen(a);i++){
                if(a[i]!=b[i])flag=1;
            }
            if(!flag){
                cout<<"YES"<<endl;
            }else{
                cout<<"NO"<<endl;
            }
        }
    }
    return 0;
}