結果

問題 No.2374 ASKT Subsequences
ユーザー hibit_athibit_at
提出日時 2023-06-16 20:09:00
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 4,101 bytes
コンパイル時間 1,279 ms
コンパイル使用メモリ 123,184 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-06-29 22:36:01
合計ジャッジ時間 21,469 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 4 ms
5,248 KB
testcase_01 AC 6 ms
5,376 KB
testcase_02 AC 3 ms
5,376 KB
testcase_03 AC 7 ms
5,376 KB
testcase_04 AC 10 ms
5,376 KB
testcase_05 AC 5 ms
5,376 KB
testcase_06 AC 7 ms
5,376 KB
testcase_07 AC 47 ms
5,376 KB
testcase_08 AC 19 ms
5,376 KB
testcase_09 AC 27 ms
5,376 KB
testcase_10 AC 121 ms
5,376 KB
testcase_11 AC 323 ms
5,376 KB
testcase_12 AC 193 ms
5,376 KB
testcase_13 AC 162 ms
5,376 KB
testcase_14 AC 442 ms
5,376 KB
testcase_15 AC 563 ms
5,376 KB
testcase_16 AC 379 ms
5,376 KB
testcase_17 AC 811 ms
5,376 KB
testcase_18 AC 1,203 ms
5,376 KB
testcase_19 AC 1,116 ms
5,376 KB
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 AC 1,200 ms
5,376 KB
testcase_24 AC 1,206 ms
5,376 KB
testcase_25 AC 118 ms
5,376 KB
testcase_26 TLE -
testcase_27 AC 232 ms
5,376 KB
testcase_28 AC 421 ms
5,376 KB
testcase_29 AC 1,867 ms
5,376 KB
testcase_30 AC 783 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <algorithm>
#include <iostream>
#include <iomanip>
#include <limits.h>
#include <map>
#include <math.h>
#include <numeric>
#include <queue>
#include <set>
#include <sstream>
#include <string>
#include <utility>
#include <vector>
#include <stack>
#include <complex>
#include <cassert>

using namespace std;
#define rep(i, n) for (int i = 0; i < n; i++)
#define rep1(i, n) for (int i = 1; i < n + 1; i++)
#define rev(i, n) for (int i = n - 1; i >= 0; i--)
#define all(A) A.begin(), A.end()
#define itr(A, l, r) A.begin() + l, A.begin() + r
#define debug(var) cout << #var << " = " << var << endl;
typedef long long ll;

template <typename T1, typename T2>
ostream &operator<<(ostream &os, const pair<T1, T2> &p)
{
    os << "(" << p.first << "," << p.second << ")";
    return os;
}

template <typename T1, typename T2>
istream &operator>>(istream &is, pair<T1, T2> &p)
{
    is >> p.first >> p.second;
    return is;
}

template <typename T>
ostream &operator<<(ostream &os, const vector<T> &v)
{
    for (int i = 0; i < (int)v.size(); i++)
    {
        os << v[i] << (i + 1 != (int)v.size() ? " " : "");
    }
    return os;
}

template <typename T>
ostream &operator<<(ostream &os, const vector<vector<T>> &v)
{
    for (int i = 0; i < (int)v.size(); i++)
    {
        os << v[i] << endl;
    }
    return os;
}

template <typename T>
ostream &operator<<(ostream &os, const vector<vector<vector<T>>> &v)
{
    for (int i = 0; i < (int)v.size(); i++)
    {
        os << "i = " << i << endl;
        os << v[i];
    }
    return os;
}

template <typename T>
istream &operator>>(istream &is, vector<T> &v)
{
    for (T &in : v)
        is >> in;
    return is;
}

template <typename T, typename S>
ostream &operator<<(ostream &os, map<T, S> &mp)
{
    for (auto &[key, val] : mp)
    {
        os << key << ":" << val << " ";
    }
    cout << endl;
    return os;
}

template <typename T>
ostream &operator<<(ostream &os, set<T> st)
{
    auto itr = st.begin();
    for (int i = 0; i < (int)st.size(); i++)
    {
        os << *itr << (i + 1 != (int)st.size() ? " " : "");
        itr++;
    }
    return os;
}

template <typename T>
ostream &operator<<(ostream &os, multiset<T> st)
{
    auto itr = st.begin();
    for (int i = 0; i < (int)st.size(); i++)
    {
        os << *itr << (i + 1 != (int)st.size() ? " " : "");
        itr++;
    }
    return os;
}

template <typename T>
ostream &operator<<(ostream &os, queue<T> q)
{
    while (q.size())
    {
        os << q.front() << " ";
        q.pop();
    }
    return os;
}

template <typename T>
ostream &operator<<(ostream &os, deque<T> q)
{
    while (q.size())
    {
        os << q.front() << " ";
        q.pop_front();
    }
    return os;
}

template <typename T>
ostream &operator<<(ostream &os, stack<T> st)
{
    while (st.size())
    {
        os << st.top() << " ";
        st.pop();
    }
    return os;
}

template <typename T>
ostream &operator<<(ostream &os, priority_queue<T> pq)
{
    while (pq.size())
    {
        os << pq.top() << " ";
        pq.pop();
    }
    return os;
}

template <typename T>
ostream &operator<<(ostream &os, priority_queue<T, vector<T>, greater<T>> mpq)
{
    while (mpq.size())
    {
        os << mpq.top() << " ";
        mpq.pop();
    }
    return os;
}

int main()
{
    int n;
    cin >> n;
    bool n_check = 4 <= n && n <= 2000;
    assert(n_check && "Error");
    vector<int> a(n);
    rep(i, n)
    {
        cin >> a[i];
        bool a_check = 1 <= a[i] && a[i] <= 2000;
        assert(a_check && "Error");
    }
    ll ans = 0;
    for (int d = 1; d < 1e3; d++)
    {
        // debug(d);
        vector<vector<ll>> dp(n, vector<ll>(4));
        map<pair<int, int>, int> mp;
        rep(i, n)
        {
            dp[i][0]++;
            dp[i][1] += mp[{a[i] - 10 - d, 0}];
            dp[i][2] += mp[{a[i] + d, 1}];
            dp[i][3] += mp[{a[i] - d - 1, 2}];
            rep(j, 4)
            {
                mp[{a[i], j}] += dp[i][j];
            }
            ans += dp[i][3];
        }
        // debug(mp);
        // debug(dp);
    }
    cout << ans << endl;
}
0