Skip to main content

C++ of the Day #21 - multi key sorting

Question

이번 문제는 sorting에 관한 내용입니다. 먼저 아래 코드는 Data의 vector를 sorting하는 코드로 Data의 operator<()을 사용하고 있습니다.
using namespace std;
using namespace boost::lambda;

struct Data
{
  string node;
  int id;
  Data(const string& n, int i) : node(n), id(i) {}
};

bool operator<(const Data& lhs, const Data& rhs)
{
  return lhs.node < rhs.node;
}

ostream& operator<<(ostream& os, const Data& data)
{
  os << setw(5) << left << data.node << " : " << setw(2) << right << data.id;
  return os;
}

int main()
{
  vector<data> vs;
  vs.push_back(Data("CAN2", 2));
  vs.push_back(Data("CAN10", 13));
  vs.push_back(Data("CAN10", 10));
  vs.push_back(Data("CAN1", 1));
  vs.push_back(Data("CAN11", 11));

  stable_sort(vs.begin(), vs.end());

  for_each(vs.begin(), vs.end(), cout << _1 << "n");
}
하지만 위의 코드를 실행시키면 다음과 같이 원하는대로 출력되지 않습니다. (여기서 원하는 순서는 CAN1 -> CAN2 -> CAN10 -> CAN11 이라고 가정하겠습니다.)
CAN1  :  1
CAN10 : 13
CAN10 : 10
CAN11 : 11
CAN2  :  2
그럼 이 코드를 원하는 대로 출력하려면 어떻게 operator<()의 내용을 바꾸어야 할까요? 단 문자열 앞의 세자리는 항상 "CAN"입니다.


Answer

어렵게 하자면 문자열의 앞에서부터 세자리 떼고 atoi로 integer 타입으로 변경한 후 비교하는 방법등 끝도 없겠죠? 여기서는 간단히 문자열의 길이를 sorting 키에 추가하는 방법을 사용하겠습니다. 즉, 문자열의 길이가 짧을수록, 그리고 string의 사전적 순서가 앞일수록 먼저 출력되도록 합니다. 그냥 원시적(?)으로 구현하면 다음과 같습니다.
if (lhs.node.length() == rhs.node.length())
  return lhs.node < rhs.node;
return lhs.node.length() < rhs.node.length();
하지만 std::pair를 사용하면 더욱 쉽게 함수를 구현할 수 있습니다. std::pair는 자신의 first와 second에 대해 사전적 순서를 가지고 구현된 operator<()을 가지고 있기 때문에 이런 구현이 가능합니다.
return make_pair(lhs.node.length(), lhs.node) <
make_pair(rhs.node.length(), rhs.node);
다시 컴파일하여 실행시킨 출력은 다음과 같습니다.
CAN1  :  1
CAN2  :  2
CAN10 : 13
CAN10 : 10
CAN11 : 11
출력을 보니 여기에 다시 id를 키로 넣어 마저 정렬시키고 싶은 생각이 드는군요. ((이 예제를 위해 sort 대신 stable_sort를 사용했답니다. ;-))) 하지만 pair는 두 개의 데이터만을 가질 수 있으므로 사용이 불가능하죠. 여기가 boost::tuple이 등장하는 시간입니다. boost::tuple를 사용하여 간단히 다음과 같이 구현이 가능합니다. ((tuple을 비교하기 위해서는 tuple.hpp 외에 tuple_comparison.hpp 파일도 #include해주어야 합니다.)) boost::tuple 역시 자신의 데이터들에 대해 사전적 순서를 가지고 구현된 operator<()을 가지고 있기 때문에 이런 구현이 가능합니다.
return make_tuple(lhs.node.length(), lhs.node, lhs.id) < 
make_tuple(rhs.node.length(), rhs.node, rhs.id);
드디어 출력이 완성되었네요. ;-)
CAN1  :  1
CAN2  :  2
CAN10 : 10
CAN10 : 13
CAN11 : 11

Comments

Popular posts from this blog

1의 개수 세기

저도 간단한 알고리즘 문제 하나... :-)

어떤 수 n이 주어졌을때 1~n까지의 수를 쭈욱 썼을때 나오는 1의 개수를 구하는 문제입니다.

예를 들어 13이라는 수가 주어지면 1~13까지의 수 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13에서 1은 1, 10, 11, 12, 13에 나오며 그 개수는 6이 됩니다. 즉, f(13)=6.

원래 문제는 f(n)=n이 되는 1이 아닌 가장 작은 수를 구하는 문제인데 이 문제의 경우에는 처음부터 쭈욱 세어나가면 되기 때문에 간단히 다음과 같이 구현을 하면 됩니다. ((한가지 주의할 점은 이전에 찾았던 n-1값을 사용하지 않고 다시 처음부터 n까지 값을 계산하면 시간이 너무 많이 걸린다는 점입니다. 위의 코드에서는 static 변수를 사용하여 이전 값에 계속 더해나가는 방법을 사용했습니다.))


#include

int count1(int n)
{
static int cnt = 1; // not 0 because n starts from 2. see main.

while (n > 0) {
if ((n % 10) == 1) ++cnt;
n /= 10;
}

return cnt;
}

int main()
{
using namespace std;

int n = 2;

while (count1(n) != n) ++n;
cout << n << endl;
}


좀 재미가 없죠? 그래서 이번 문제는 어떤 수 n에 대해서 f(n)을 O(1)시간에 구하는 알고리즘을 만드는 것입니다. 관심있으신 분들은 한번 풀어보세요. 제가 만든 코드는 내일 올려보겠습니다.

C++ of the Day #9 - Boost.Python 사용하기 #1

Python은 가장 인기있는 interpret 언어중의 하나입니다. Python의 장점 중 하나는 C/C++ 모듈과 쉽게 연동할 수 있다는 점입니다. 물론 손으로 일일히 wrapper를 만드는 것은 손이 많이 가고 에러를 만들수 있는 작업이나 SWIG등과 같은 도구를 사용하면 쉽게 python 모듈을 만들 수 있습니다.

Boost.Python은 이런 SWIG와 같이 python 모듈을 쉽게 만들 수 있도록 도와주는 라이브러리로 순수 C++만을 사용한다는 점이 SWIG와 다른 점입니다. 그리고 개인적으로는 Boost 라이브러리에 포함되어 있는 것들이 왠지 좀 더 믿음직스러워서... :-)

이번 글에서는 Boost.Python 문서에 나와 있는 예제를 가지고 간단하게 python 모듈을 만드는 방법에 대해서 알아보겠습니다.

Requirements리눅스
이 글에서는 리눅스 환경에서의 사용 방법을 설명한다.Boost.Python 라이브러리 (1.33.1)
Boost 라이브러리를 다운로드받아 아래와 유사한 명령으로 라이브러리를 빌드한다.
bjam -sTOOLS=gcc -with-python install

bjam의 --prefix 옵션으로 라이브러리가 설치될 위치를 변경할 수 있다.Python 라이브러리 (2.4.3)
Python을 다운로드 받아 빌드하여 설치한다.
위의 경우와 유사하게 configure의 --prefix 옵션으로 설치될 위치를 변경할 수 있다.

Write C++ Code다음과 같이 코드를 작성한다.

// greet.cpp #include <stdexcept> char const* greet(unsigned x) { static char const* const msgs[] = { "hello", "Boost.Python", "world!" }; if (x > 2) throw std::range_error("greet: index out of range"…

Hello Wordpress, again.

한 두주일 정도 Textpattern을 사용해봤는데 다시 Wordpress로 돌아오기로 결정했습니다. 무엇보다 스킨 변경이 너무 복잡하고 사용자층이 Wordpress에 비해 너무 앏네요. 원하는 plugin도 찾기 어렵고... :-|

그동안 Textpattern에 썼던 글들은 모두 Wordpress로 옮겼습니다. 2개 있던 댓글도 옮겼는데 그중의 하난 제가 쓴... ;-)

애초에 wp-dokuwiki plugin이 무거워서 옮겼던 것이라 이 plugin은 설치를 안할 예정인데 몇가지 아쉬운 점이 있네요.

첫째는 code highlighting 기능인데 이 기능은 예전에 만들어 놨던 것을 조금 수정해서 쓰려고 준비중입니다. 두번째는 Footnote 기능인데 찾아보니 Footnotes 0.9 Plugin for WordPress 2.0.x라는게 있네요.

이정도면 비록 wiki syntax에 비할바는 아니지만 쓸만할 것 같습니다. :-)