C++で同じ連続した違いがある番号

連続する2桁ごとの絶対差がKになるように、長さNのすべての非負の整数を見つける必要があるとします。回答のすべての数値は、数値0自体を除いて先行ゼロを持ってはならないことに注意する必要があります。回答は任意の順序で返送する場合があります。したがって、N=3およびK=7の場合、出力は[181,292,707,818,929]になります。ここでは、先行ゼロが1つあるため、070は有効な数値ではないことがわかります。

これを解決するには、次の手順に従います-

• dpと呼ばれる1つの行列を作成し、そのサイズはn + 1になり、1から9をdp [1]

  に入力します。

• 1からN–1の範囲のiの場合

  • 訪問済みと呼ばれるセットを定義する

  • 0からdp[i]

    のサイズの範囲のjの場合

    • x：=dp [i、j]

    • lastNum：=xの最後の桁

    • 数字：=lastNum + k

    • 数字が0から9の範囲にあり、（x * 10 +数字）が訪問されていない場合、

      • （10 * x +桁）をdp [i + 1]

        に挿入します

      • 訪問した配列に10*x+桁を挿入

    • 数字：=lastNum – K

    • 数字が0から9の範囲にあり、（x * 10 +数字）が訪問されていない場合、

      • （10 * x +桁）をdp [i + 1]

        に挿入します

      • 訪問した配列に10*x+桁を挿入

• Nが1の場合、dp [N]

  に0を挿入します。

• dp [N]

  を返します

理解を深めるために、次の実装を見てみましょう-

例

#include <bits/stdc++.h>
using namespace std;
void print_vector(vector<int> v){
   cout << "[";
   for(int i = 0; i<v.size(); i++){
      cout << v[i] << ", ";
   }
   cout << "]"<<endl;
}
class Solution {
   public:
   vector<int> numsSameConsecDiff(int N, int K) {
      vector <int> dp[N + 1];
      for(int i = 1; i <= 9; i++){
         dp[1].push_back(i);
      }
      for(int i = 1; i < N; i++){
         set <int> visited;
         for(int j = 0; j < dp[i].size(); j++){
            int x = dp[i][j];
            int lastNum = x % 10;
            int digit = lastNum + K;
            if(digit >= 0 && digit <= 9 && !visited.count(x * 10 + digit)){
               dp[i + 1].push_back(x * 10 + digit);
               visited.insert(x * 10 + digit);
            }
            digit = lastNum - K;
            if(digit >= 0 && digit <= 9 && !visited.count(x * 10 + digit)){
               dp[i + 1].push_back(x * 10 + digit);
               visited.insert(x * 10 + digit);
            }
         }
      }
      if(N == 1){
         dp[N].push_back(0);
      }
      return dp[N];
   }
};
main(){
   Solution ob;
   print_vector(ob.numsSameConsecDiff(3,7));
}

入力

3
7

出力

[181,292,707,818,929]