2028 자기복제수

2028번: 자기복제수 (acmicpc.net)

2028번: 자기복제수

#define _CRT_SECURE_NO_WARNINGS
#include<iostream>
#include<string>
#include<cstring>


using namespace std;
int main() {
	cin.tie(NULL);
	cin.sync_with_stdio(false);
	int num;
	
	int fornum = 0;
	cin >> fornum;
	for (int z = 0; z < fornum; z++) {
		cin >> num;
		int cnt = 0;
		int numcopy1 = num;
		int numcopy2 = num;
		while (numcopy1 != 0) {
			numcopy1 /= 10;
			cnt++;
		}


		int* narr = new int[cnt];
		for (int i = cnt - 1; i >= 0; i--) {

			narr[i] = num % 10;
			num /= 10;

		}

		int mnum = numcopy2 * numcopy2;
		int mcnt = 0;
		int mnumcopy1 = mnum;
		while (mnumcopy1 != 0) {
			mnumcopy1 /= 10;
			mcnt++;

		}

		int* marr = new int[mcnt];
		for (int i = mcnt - 1; i >= 0; i--) {

			marr[i] = mnum % 10;
			mnum /= 10;
	
		}

		int flag = 0;
		for (int i = 0; i < cnt; i++) {
			if (marr[mcnt - cnt + i] != narr[i]) {
				flag = 1;
			
			};
		}

		if (flag == 0) {
			cout << "YES\n";
		}
		else {
			cout << "NO\n";
		}
		
	}
}