hdu 2054 A == B?

http://acm.hdu.edu.cn/showproblem.php?pid=2054
此题巨坑，刚开始我以为是简单的水题，就用strcmp过，
but错了，后来经过我苦思冥想，结果还有几组数据
0.0 和 0，1.000和1.0 ， 但是我不太确定前面的0是不是有作用我还是写了，但是有人过的时候，前面的0没考虑比如：
002和2可能是相等的，也可能是不想等的所以不用判断，只能说明hdu数据不是很强啊，嘿嘿
代码如下：

#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;
char a[500005];
char b[500005];
int main()
{
    while(cin>>a>>b)
    {
        int len1=strlen(a);
        int len2=strlen(b);
        bool flag=0, k1=0, k2=0;
        int i,j=0;
        for(i=0; i<len1; i++)//判断前面0
        {
            if(a[i]!='0' && !flag)
               flag=1;
            if(flag)
               a[j++]=a[i];
            if(a[i] == '.')
                k1=1;
        }
        a[j]='\0';
        flag=0,j=0;
        for(i=0; i<len2; i++)//判断前面0
        {
            if(b[i]!='0' && !flag)
               flag=1;
            if(flag)
                b[j++]=b[i];
            if(b[i] == '.')
                k2=1;
        }
        b[j]='\0';//结束啊
        len1=strlen(a);
        len2=strlen(b);
        for(i=len1-1; a[i]!='.' && k1; i--)//判断后面的啊
        {
            if(a[i]=='0' || a[i]=='.')
                a[i]='\0';
            else
                k1=0;
        }
        if(k1)//如果没变的话
            a[i]='\0';
        for(i=len2-1; b[i]!='.'&&k2; i--)//判断后面的
        {
            if(b[i]=='0' || b[i]=='.')
               b[i]='\0';
            else
               k2=0;
        }
        if(k2)
            b[i]='\0';
        if(strcmp(a,b)==0)
            puts("YES");
        else
            puts("NO");
    }
    return 0;
}