기본 콘텐츠로 건너뛰기

10989 - 수 정렬하기3 (Radix Sort) [C++]

기수 정렬로 풀어보라해서 기수 정렬로 풀었는데.. 메모리 초과가 계속 떠서 알아봤더니 
기수 정렬로는 못 풀고 카운팅 정렬로만 풀 수 있다고 한다. 카운팅 정렬로 풀어봐야겠다.

#include <iostream>
#include <vector>

using namespace std;

class Radix
{
private:
 vector<int> A, B, C, D, E;
 vector<vector<int>> Bucket;
 int N;
public:
 Radix(int n);
 void Input();
 void Sort();
 void Print();
};

int main()
{
 cin.tie(NULL);

 int N;
 cin >> N;

 Radix R(N);
 R.Input();
 R.Sort();
 R.Print();
}

Radix::Radix(int n)
{
 this->N = n;
 
 for (int i = 0; i < 10; i++)
 {
  vector<int> Temp;
  Temp.push_back(i);
  Bucket.push_back(Temp);
 }
}

void Radix::Input()
{
 for (int i = 0; i < this->N; i++)
 {
  int input;
  cin >> input;

  if (input < 10)
   this->A.push_back(input);
  else if (input < 100)
   this->B.push_back(input);
  else if (input < 1000)
   this->C.push_back(input);
  else if (input < 10000)
   this->D.push_back(input);
  else
   this->E.push_back(input);
 }
}

void Radix::Sort()
{
 // AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
 for (int i = 0; i < A.size(); i++)
  Bucket[A[i] % 10].push_back(A[i]);

 A.clear();
 
 for (int i = 0; i < Bucket.size(); i++)
 {
  for (int j = 1; j < Bucket[i].size(); j++)
   A.push_back(Bucket[i][j]);

  Bucket[i].clear();
 }

 // BBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBB
 for (int i = 0; i < B.size(); i++)
  Bucket[B[i] % 10].push_back(B[i]);

 B.clear();

 for (int i = 0; i < Bucket.size(); i++)
 {
  for (int j = 0; j < Bucket[i].size(); j++)
   B.push_back(Bucket[i][j]);

  Bucket[i].clear();
 }

 for (int i = 0; i < B.size(); i++)
  Bucket[(B[i] % 100) / 10].push_back(B[i]);

 B.clear();

 for (int i = 0; i < Bucket.size(); i++)
 {
  for (int j = 0; j < Bucket[i].size(); j++)
   B.push_back(Bucket[i][j]);

  Bucket[i].clear();
 }

 // CCCCCCCCCCCCCCCCCCCCCCCCCCCCCCCCCC
 for (int i = 0; i < C.size(); i++)
  Bucket[C[i] % 10].push_back(C[i]);

 C.clear();

 for (int i = 0; i < Bucket.size(); i++)
 {
  for (int j = 0; j < Bucket[i].size(); j++)
   C.push_back(Bucket[i][j]);

  Bucket[i].clear();
 }

 for (int i = 0; i < C.size(); i++)
  Bucket[(C[i] % 100) / 10].push_back(C[i]);

 C.clear();

 for (int i = 0; i < Bucket.size(); i++)
 {
  for (int j = 0; j < Bucket[i].size(); j++)
   C.push_back(Bucket[i][j]);

  Bucket[i].clear();
 }

 for (int i = 0; i < C.size(); i++)
  Bucket[(C[i] % 1000) / 100].push_back(C[i]);

 C.clear();

 for (int i = 0; i < Bucket.size(); i++)
 {
  for (int j = 0; j < Bucket[i].size(); j++)
   C.push_back(Bucket[i][j]);

  Bucket[i].clear();
 }

 // DDDDDDDDDDDDDDDDDDDDDDDDDDDDDDDDDD
 for (int i = 0; i < D.size(); i++)
  Bucket[D[i] % 10].push_back(D[i]);

 D.clear();

 for (int i = 0; i < Bucket.size(); i++)
 {
  for (int j = 0; j < Bucket[i].size(); j++)
   D.push_back(Bucket[i][j]);

  Bucket[i].clear();
 }

 for (int i = 0; i < D.size(); i++)
  Bucket[(D[i] % 100) / 10].push_back(D[i]);

 D.clear();

 for (int i = 0; i < Bucket.size(); i++)
 {
  for (int j = 0; j < Bucket[i].size(); j++)
   D.push_back(Bucket[i][j]);

  Bucket[i].clear();
 }

 for (int i = 0; i < D.size(); i++)
  Bucket[(D[i] % 1000) / 100].push_back(D[i]);

 D.clear();

 for (int i = 0; i < Bucket.size(); i++)
 {
  for (int j = 0; j < Bucket[i].size(); j++)
   D.push_back(Bucket[i][j]);

  Bucket[i].clear();
 }

 for (int i = 0; i < D.size(); i++)
  Bucket[(D[i] % 10000) / 1000].push_back(D[i]);

 D.clear();

 for (int i = 0; i < Bucket.size(); i++)
 {
  for (int j = 0; j < Bucket[i].size(); j++)
   D.push_back(Bucket[i][j]);

  Bucket[i].clear();
 }
}

void Radix::Print()
{
 for (int i = 0; i < A.size(); i++)
  cout << A[i] << "\n";
 for (int i = 0; i < B.size(); i++)
  cout << B[i] << "\n";
 for (int i = 0; i < C.size(); i++)
  cout << C[i] << "\n";
 for (int i = 0; i < D.size(); i++)
  cout << D[i] << "\n";
 for (int i = 0; i < E.size(); i++)
  cout << E[i] << "\n";
}

댓글

이 블로그의 인기 게시물

11401 : 이항계수 3 (Dynamic Programming, Divide and Conquer) [C++]

1번, 2번 문제들과 확연히 차이나는 입력의 범위. 400만 ! DP를 사용해서 풀 수 없는 문제이다 . 하지만 DP가 쓰이긴 한다! 수학은 너무 어렵다. 곱셈의 역원을 공부해보다가 모르겠어서 도움을 구했다. 왜 곱셈의 역원을 구해야하는가? N! / (K! * (N-K)!) 에서 K! * (N-K)! 의 역원을 구해야 하기 때문! 곱셈의 역원을 구하는 정리인 페르마의 소정리를 이용하면 p가 1000000007 이지만 분할 정복을 이용한 제곱 수 계산 덕분에 logP 시간 소요. 분할 정복을 이용한 제곱 수 계산은 계속 써먹을 것 같아서 따로 올려놓았다. DP가 쓰이는 부분은 구해준 400만의 역원을 바탕으로 모든 역원을 구하는 부분이다. 그러므로 총 시간 소요는 O(N+LogP) long long BinomialCoefficient :: GetNum ( int N , int K ) { this - > Factorial [ 1 ] = 1 ; for ( int i = 2 ; i < = 4000000 ; i + + ) this - > Factorial [ i ] = ( this - > Factorial [ i - 1 ] * i ) % P ; this - > Invert [ 4000000 ] = this - > Pow_DC ( this - > Factorial [ 4000000 ] , P - 2 ) ; for ( int i = 4000000 - 1 ; i > 0 ; i - - ) this - > Invert [ i ] = ( this - > Invert [ i + 1 ] * ( i + 1 ) ) % P ; if ( N = = K | | K = ...

1978 : 소수 찾기 [C++]

# include < iostream > # include < vector > using namespace std ; int main ( ) { cin . tie ( NULL ) ; vector < int > Primes ; Primes . push_back ( 2 ) ; Primes . push_back ( 3 ) ; for ( int i = 4 ; i < 1000 ; i + + ) { bool IsPrime = true ; if ( i % 2 = = 0 | | i % 3 = = 0 ) continue ; for ( int j = 4 ; j < i ; j + + ) { if ( i % j = = 0 ) { IsPrime = false ; break ; } } if ( IsPrime ) Primes . push_back ( i ) ; } int N , Count = 0 ; cin > > N ; for ( int i = 0 ; i < N ; i + + ) { int Input ; cin > > Input ; for ( int j = 0 ; j < Primes . size ( ) ; j + + ) if ( Input = = Primes [ j ] ) Count + + ; } cout < < Count < < " \n " ; return 0 ; }