C++の配列内のK-diffペア

配列と整数kがあるとすると、配列内の一意のk-diffペアの数を見つける必要があります。ここで、k-diffペアは（i、j）のようになります。ここで、iとjは両方とも配列に存在し、それらの絶対差はkです。

したがって、入力が[3,1,4,1,5]、k =2の場合、配列のような（1,3）と（の2つの2差分ペアがあるため、出力は2になります。 3,5）。

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

• 見た、完了したと呼ばれるマップを定義する

• 1つのセットを定義する

• k <0の場合、-

  • 0を返す

• 初期化i：=0の場合、i

  • （seen [nums [i]]を1増やします）

  • nums[i]をs

    に挿入します

• ans：=0

• s内の要素ごとに、次のようにします-

  • kが0と同じ場合、-

    • 見た場合[it]>1、次に-

      • （ansを1増やします）

  • それ以外の場合

    • done[it]を1増やします

    • （it + k）が表示されているが、実行されていない場合、-

      • （ansを1増やします）

    • （it --k）が表示されているが、実行されていない場合、-

      • （ansを1増やします）

• ansを返す

例

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

#include <bits/stdc++.h&g;
using namespace std;
class Solution {
public:
   int findPairs(vector<int>& nums, int k) {
      map<int, int> seen, done;
      set<int> s;
      if (k < 0)
         return 0;
      for (int i = 0; i < nums.size(); i++) {
         seen[nums[i]]++;
         s.insert(nums[i]);
      }
      int ans = 0;
      for (auto it = s.begin(); it != s.end(); it++) {
         if (k == 0) {
            if (seen[*it] > 1)
            ans++;
         }
         else {
            done[*it]++;
            if (seen.find(*it + k) != seen.end() && done.find(*it + k) == done.end())
               ans++;
            if (seen.find(*it - k) != seen.end() && done.find(*it - k) == done.end())
               ans++;
         }
      }
      return ans;
   }
};
main(){
   Solution ob;
   vector<int> v = {3,1,4,1,5};
   cout << (ob.findPairs(v, 2));
}

入力

{3,1,4,1,5}, 2

出力

2