PAT:A1060. Are They Equal (17/25)

If a machine can save only 3 significant digits, the float numbers 12300 and 12358.9 are considered equal since they are both saved as 0.123*10⁵ with simple chopping. Now given the number of significant digits on a machine and two float numbers, you are supposed to tell if they are treated equal in that machine.

Input Specification:

Each input file contains one test case which gives three numbers N, A and B, where N (<100) is the number of significant digits, and A and B are the two float numbers to be compared. Each float number is non-negative, no greater than 10¹⁰⁰, and that its total digit number is less than 100.

Output Specification:

For each test case, print in a line "YES" if the two numbers are treated equal, and then the number in the standard form "0.d₁...d_N*10^k" (d₁>0 unless the number is 0); or "NO" if they are not treated equal, and then the two numbers in their standard form. All the terms must be separated by a space, with no extra space at the end of a line.

Note: Simple chopping is assumed without rounding.

Sample Input 1:

3 12300 12358.9

Sample Output 1:

YES 0.123*10^5

Sample Input 2:

3 120 128

Sample Output 2:

NO 0.120*10^3 0.128*10^3

#include <iostream>
#include <string>
#include <string.h>
using namespace std;
int N;
string A, B;
string standard(string x) {
    if(x.find('.') > 1) {
        if(x.find('.') <= N) {
            x.erase(x.find('.'));
            //return x;
        } else {
            //return x;
        }
        x = x.substr(0, N);
        return x;

    } else {
        //cout << x.find('.') << endl;
        x.erase(0, 2);
        string::iterator it = x.begin();
        while(*(it) == '0' && it != x.end()) {
            x.erase(x.begin());
            it++ ;
        }
        //return x;
        string k;
        for(int i = 0; i < N && it != x.end(); i++,it++){
            k.push_back(*(it));
        }
        return k;
    }
}
int standardN(string x, int y) {
    string::iterator it = x.begin();
    if(x.find('.') == 1){
        //x.erase(0, 2);
        it = it + 2;
        while(*(it) == '0') {
            it++;
            //x.erase(0);
            y++;
        }
        return -y;
        //return x;
    } else {
        while(*(it) != '.' && it != x.end()) {
            //x.erase(0);
            it++;
            y++;
            //cout << y << endl;
        }
        return y;
    }
    return y;
}
int main() {
    int c1 = 0, c2 = 0;
    cin >> N >> A >> B;
    c1 = standardN(A, c1);
    c2 = standardN(B, c2);
    A = standard(A);
    B = standard(B);
    if(A == B && c1 == c2) {
        cout << "YES " << "0." << A << "*10^" << c1 << endl;
    } else {
        cout << "NO " << "0." << A << "*10^" << c1 << " " << "0." << B << "*10^" << c2 <<endl;
    }
    return 0;
}