#include <iostream>
#include <set>
#include <utility>
#include <cmath>
using namespace std;
int main() {
int n;
cin >> n;
// Używamy setu par do przechowywania unikalnych akwariów
// Para {{a,b},h} reprezentuje akwarium, gdzie {a,b} to nieuporządkowana para
set<pair<pair<int,int>,int>> aquariums;
// Przeglądamy wszystkie możliwe wymiary a, b, h
for (int a = 1; a <= n; a++) {
for (int b = 1; b <= a; b++) { // b <= a aby uniknąć duplikatów {a,b} i {b,a}
for (int h = 1; h <= n; h++) {
// Obliczamy przekątną
double diagonal = sqrt(a*a + b*b + h*h);
// Sprawdzamy czy przekątna jest liczbą całkowitą i nie przekracza n
if (diagonal == (int)diagonal && diagonal <= n) {
// Tworzymy uporządkowaną parę {mniejsze, większe} dla podstawy
pair<int,int> base = {b, a};
aquariums.insert({base, h});
}
}
}
}
// Wypisujemy liczbę różnych akwariów
cout << aquariums.size() << endl;
return 0;
}
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 | #include <iostream> #include <set> #include <utility> #include <cmath> using namespace std; int main() { int n; cin >> n; // Używamy setu par do przechowywania unikalnych akwariów // Para {{a,b},h} reprezentuje akwarium, gdzie {a,b} to nieuporządkowana para set<pair<pair<int,int>,int>> aquariums; // Przeglądamy wszystkie możliwe wymiary a, b, h for (int a = 1; a <= n; a++) { for (int b = 1; b <= a; b++) { // b <= a aby uniknąć duplikatów {a,b} i {b,a} for (int h = 1; h <= n; h++) { // Obliczamy przekątną double diagonal = sqrt(a*a + b*b + h*h); // Sprawdzamy czy przekątna jest liczbą całkowitą i nie przekracza n if (diagonal == (int)diagonal && diagonal <= n) { // Tworzymy uporządkowaną parę {mniejsze, większe} dla podstawy pair<int,int> base = {b, a}; aquariums.insert({base, h}); } } } } // Wypisujemy liczbę różnych akwariów cout << aquariums.size() << endl; return 0; } |
English