aboutsummaryrefslogtreecommitdiff
path: root/challenge-241/deadmarshal/cpp/ch-2.cpp
blob: b53ab789cb68feb71e7e3b1f10a2ac1e38aedcd1 (plain)
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
38
39
40
41
42
43
44
45
46
47
48
#include<iostream>
#include<vector>
#include<algorithm>

template<typename T>
size_t count_factors(T n)
{
  size_t count{};
  T c = 2;
  while(n > 1)
  {
    if(n % c == 0)
    {
      n /= c;
      count++;
    }
    else c++;
  }
  return count;
}

template<typename T>
void prime_order(std::vector<T> &vec)
{
  std::sort(vec.begin(),vec.end(),[&](T a, T b){
    size_t fa = count_factors<T>(a);
    size_t fb = count_factors<T>(b);
    return fa == fb ? a < b : fa < fb;
  });
}

template<typename T>
std::ostream &operator<<(std::ostream &os,
			 const std::vector<T>& vec)
{
  for(const auto &e : vec) os << e << ' ';
  os << "\n";
  return os;
}

int main()
{
  std::vector<int> vec{11,8,27,4};
  prime_order<int>(vec);
  std::cout << vec;
  return 0;
}