các giúp mình với. Làm trong c++ nha.

#include <iostream>
using namespace std;
int main()
{
int a,b,c;
cout<<"Nhap ba canh tam giac a,b,c"<<endl;
cin>>a>>b>>c;
if ( a*a + b*b == c*c || a*a + c*c == b*b || b*b + c*c == a*a )

{

cout<<"YES"<<endl;

}

else

{

cout<<"NO"<<endl;

}

return 0;

}

#include <iostream>
using namespace std;
bool check_right_triangle(int a, int b, int c) {
  return (a * a + b * b == c * c ||
          a * a + c * c == b * b ||
          b * b + c * c == a * a);
}
int main() {
  int a, b, c;
  cout << "Enter a,b,c: ";
  cin >> a >> b >> c;
  if (check_right_triangle(a, b, c)) {
    cout << "YES" << endl;
  } else {
    cout << "NO" << endl;
  }
  return 0;
}