fork download
  1. #include<bits/stdc++.h>
  2. #define ull unsigned long long
  3. #define ll long long
  4. #define all(x) x.begin(), x.end()
  5. using namespace std;
  6. const int maxn = 1e6 + 5;
  7. unordered_map<int, int> D;
  8. int sumDigit(int n){
  9. int sum = 0;
  10. while(n!=0){
  11. sum += n%10;
  12. n/=10;
  13. }
  14. return sum;
  15. }
  16. int main() {
  17. ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
  18. int n, x;
  19. cin>>n;
  20. for(int i=1;i<=n;++i){
  21. cin>>x;
  22. D[sumDigit(x)]++;
  23. }
  24. int res = 0;
  25. for(auto it:D){
  26. if(it.second > 1){
  27. res += it.second * (it.second - 1) / 2;
  28. }
  29. }
  30. cout<<res;
  31. }
  32.  
  33.  
Success #stdin #stdout 0.01s 5276KB
stdin
5
3 8 21 17 12
stdout
4