結果

問題 No.2374 ASKT Subsequences
コンテスト
ユーザー inkyaman
提出日時 2024-03-25 22:42:41
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 241 ms / 2,000 ms
コード長 995 bytes
コンパイル時間 1,098 ms
コンパイル使用メモリ 116,456 KB
最終ジャッジ日時 2025-02-20 13:49:36
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

#define _USE_MATH_DEFINES
#include <iostream>
#include <string>
#include <algorithm>
#include <vector>
#include <queue>
#include <math.h>
#include <cmath>
#include <stack>
#include <map>
#include <set>
#include <numeric>
#include <iomanip>
#include <climits>
#include <functional>
#include <cassert>
#include <tuple>
using namespace std;
using ll = long long;

int N;
int A[2020];

int main() {

    cin >> N;
    for(int i = 0; i < N; ++i) cin >> A[i];

    ll ans = 0;
    for(int k = 1; k <= 2000; ++k) {
        vector dp(2001, vector(4,0LL));
        for(int i = 0; i < N; ++i) {
            int a3 = A[i]-k-1;
            if(a3 > 0 && a3 <= 2000) dp[A[i]][3] += dp[a3][2];
            int a2 = A[i]+k;
            if(a2 > 0 && a2 <= 2000) dp[A[i]][2] += dp[a2][1];
            int a1 = A[i]-k-10;
            if(a1 > 0 && a1 <= 2000) dp[A[i]][1] += dp[a1][0];
            dp[A[i]][0]++;
        }
        for(int i = 1; i <= 2000; ++i) ans += dp[i][3];
    }

    cout << ans << endl;

}
0