1060 Are They Equal（25 分）（C++）

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​5​​ 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​100​​, 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[1]...d[N]*10^k (d[1]>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

题目大意：理解起来并不麻烦，将一个数写成科学计算法，第一位为0，保留三位有效数字，判断两个数是否相同。

注意最后一个测试点，应该是原有的字符串长度都不够n。

#include <iostream>
#include <string>
using namespace std;
int n, a, b;
string s1, s2;
int get_num(string &s){
	int result = 0, flag = 1, i = 0, cnt = 0;
	while(s[i] == '0')
		++ i;
	if(s[i] == '.'){
		flag = -1;
		++ i;
	}
	bool flag_cnt = flag == -1 ? false : true;
	while(i < s.length() && isdigit(s[i])){
		if(flag == -1 && s[i] != '0'){
			flag = 0;
			flag_cnt = true;
		}
		result += flag;
		if(flag_cnt == true)
			s[cnt++] = s[i++];
		else
			++ i;
	}
	if(cnt == 0)
	  result = 0;
	if(i < s.length() && s[i] == '.')
		++ i;
	while(cnt < n){
		if(cnt == s.length())
			s += '0';
		else if(i == s.length())
			s[cnt++] = '0';
		else
			s[cnt++] = s[i++];
	}
	s.resize(n);
	return result;
}
int main(){
	cin >> n >> s1 >> s2;
	a = get_num(s1);
	b = get_num(s2);
	if(a == b && s1 == s2)
		printf("YES");
	else
		printf("NO 0.%s*10^%d", s1.c_str(),a);
	printf(" 0.%s*10^%d",s2.c_str(),b);
}