結果

問題 No.3456 Common Difference is D
コンテスト
ユーザー tnakao0123
提出日時 2026-02-28 17:31:49
言語 C++17
(gcc 15.2.0 + boost 1.89.0)
コンパイル:
g++-15 -O2 -lm -std=c++17 -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
AC  
実行時間 172 ms / 2,000 ms
コード長 888 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 773 ms
コンパイル使用メモリ 75,612 KB
実行使用メモリ 8,960 KB
最終ジャッジ日時 2026-02-28 17:31:52
合計ジャッジ時間 3,278 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

/* -*- coding: utf-8 -*-
 *
 * 3456.cc:  No.3456 Common Difference is D - yukicoder
 */

#include<cstdio>
#include<map>
#include<algorithm>

using namespace std;

/* constant */

const int MAX_N = 100000;

/* typedef */

using ll = long long;
using mii = map<int,int>;

/* global variables */

int as[MAX_N];

/* subroutines */

/* main */

int main() {
  int n, d;
  scanf("%d%d", &n, &d);
  for (int i = 0; i < n; i++) scanf("%d", as + i);

  mii lacs, racs;
  for (int i = 0; i < n; i++) racs[as[i]]++;

  ll sum = 0;
  for (int i = 0; i < n; i++) {
    auto mit = racs.find(as[i]);
    if (--(mit->second) == 0) racs.erase(mit);
    
    auto lmit = lacs.find(as[i] - d);
    if (lmit != lacs.end()) {
      auto rmit = racs.find(as[i] + d);
      if (rmit != racs.end())
	sum += (ll)lmit->second * rmit->second;
    }

    lacs[as[i]]++;
  }

  printf("%ld\n", sum);

  return 0;
}
0