C++ - ランク付け(同順位は中央順位法で)!

更新日時:


前回、 C++ で複数の整数入力値にランクを付ける処理(同順位考慮)を実装してみましたが、今回はその発展形として、同順位(タイ)がある場合は中央(平均)順位(mid-rank)法で順位を付けるようにしてみました。

0. 前提条件

  • Debian GNU/Linux 10.3 (64bit) での作業を想定。
  • GCC 9.2.0 (G++ 9.2.0) (C++17) でのコンパイルを想定。

1. ソースコードの作成

  • 整数限定。(微改修で小数点数対応化)
  • 数値が大きい順に順位を付ける。
  • 同順位を考慮する。(例:要素が 2, 1, 3, 2 で、大きい順に順位付ける場合の順位を 2.5, 4, 1, 2.5 とする)

File: rank_2.cpp

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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
/***************************************************************
  Rank of numbers (integer) (by central rank method)

    DATE          AUTHOR          VERSION
    2020.09.04    mk-mode.com     1.00 新規作成

  Copyright(C) 2020 mk-mode.com All Rights Reserved.
***************************************************************/
#include <algorithm>  // for count
#include <iostream>   // for cout
#include <regex>      // for regex_search

namespace funcs {
  bool is_int(std::string s) {
    std::smatch m;
    std::regex re("^[+-]?\\d+$");

    try {
      if (!std::regex_search(s, m, re)) return false;
    } catch (std::regex_error& e) {
      return false;
    }

    return true;
  }

  bool rank_int(std::vector<int>& ns, std::vector<double>& rs) {
    unsigned int s;
    unsigned int i;
    unsigned int j;
    unsigned int c;

    try {
      s = ns.size();
      for (i = 0; i < s; i++) {
        c = 0;
        for (j = 0; j < s; j++) {
          if (ns[i] < ns[j]) c++;
        }
        rs.push_back(c + 1);
      }
    } catch (...) {
      return false;
    }

    return true;
  }

  bool mid_rank(std::vector<int>& ns, std::vector<double>& rs) {
    std::vector<double> rs_w;  // 作業用
    unsigned int i;
    unsigned int j;
    unsigned int c;
    double s;

    try {
      // 一旦、通常のランク付け
      if (!funcs::rank_int(ns, rs)) {
        std::cout << "[ERROR] Failed to rank!" << std::endl;
        return EXIT_FAILURE;
      }
      // ランク用 vector を作業にコピー
      std::copy(rs.begin(), rs.end(), std::back_inserter(rs_w));
      // 中央(平均)順位法(mid-rank)でランク付け
      rs.clear();
      for (i = 0; i < rs_w.size(); i++) {
        c = std::count(rs_w.begin(), rs_w.end(), rs_w[i]);
        s = 0;
        for (j = rs_w[i]; j < rs_w[i] + c; j++) s += j;
        rs.push_back(s / c);
      }
    } catch (...) {
      return false;
    }

    return true;
  }

  void display(std::vector<int>& ns, std::vector<double>& rs) {
    std::vector<int>::iterator it_n;     // イテレータ
    std::vector<double>::iterator it_r;  // イテレータ

    try {
      std::cout << "---" << std::endl;
      std::cout << "VALS = [";
      for (it_n = ns.begin(); it_n != ns.end(); ++it_n) {
        std::cout << *it_n;
        if (it_n != ns.end() - 1) std::cout << ", ";
      }
      std::cout << "]" << std::endl;
      std::cout << "RANK = [";
      for (it_r = rs.begin(); it_r != rs.end(); ++it_r) {
        std::cout << *it_r;
        if (it_r != rs.end() - 1) std::cout << ", ";
      }
      std::cout << "]" << std::endl;
    } catch (...) {
      throw;
    }
  }
}

int main(int argc, char* argv[]) {
  std::string buf;         // 入力バッファ
  std::vector<int> ns;     // 入力値
  std::vector<double> rs;  // ランク

  try {
    // 値入力
    while (true) {
      std::cout << "n? ";
      getline(std::cin, buf);
      if (buf.empty()) break;
      if (!funcs::is_int(buf)) {
        std::cout << "NOT INTEGER !!" << std::endl;
        continue;
      }
      ns.push_back(stoi(buf));
    }
    // ランク計算
    if (!funcs::mid_rank(ns, rs)) {
      std::cout << "[ERROR] Failed to rank!" << std::endl;
      return EXIT_FAILURE;
    }
    // 結果出力
    funcs::display(ns, rs);
  } catch (...) {
    std::cerr << "EXCEPTION!" << std::endl;
    return EXIT_FAILURE;
  }

  return EXIT_SUCCESS;
}

2. ソースコードのコンパイル

$ g++ -std=c++17 -Wall -O2 --pedantic-errors -o rank_2 rank_2.cpp

3. 動作確認

  • 空エンターで、入力終了&計算。
$ ./rank
n? 1
n? -4
n? 5
n? -2
n? a
NOT INTEGER !!
n? 5.6
NOT INTEGER !!
n? 5
n? 10
n? -8
n?
---
VALS = [1, -4, 5, -2, 5, 10, -8]
RANK = [4, 6, 2.5, 5, 2.5, 1, 7]

今回は整数限定でしたが、実数に拡張することもそう難しくはないでしょう。

以上。

 

Sponsored Link

 

コメントする