POJ 3728 Blah數集

Blah數集

#include<iostream>
#include<vector>
using namespace std;
int main()
{
  int n, i;
  unsigned a, p[51], q[51];
  vector<unsigned> v[51];
  while (cin >> a >> n)
  {
    if (v[a].empty())
    {
      v[a].push_back(a);
      p[a] = 0;
      q[a] = 0;
    }
    i = n - v[a].size();
    if (i < 0) i = 0;
    while (i--)
    {
      if (2 * v[a][p[a]] < 3 * v[a][q[a]])
      {
        v[a].push_back(2 * v[a][p[a]] + 1);
        ++p[a];
      }
      else if (2 * v[a][p[a]] > 3 * v[a][q[a]])
      {
        v[a].push_back(3 * v[a][q[a]] + 1);
        ++q[a];
      }
      else
      {
        v[a].push_back(3 * v[a][q[a]] + 1);
        ++q[a];
        ++p[a];
      }
    }
    cout << v[a][n-1] << endl;
  }
  return 0;
}