結果

問題 No.1623 三角形の制作
ユーザー tokusakuraitokusakurai
提出日時 2021-07-23 21:30:44
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 58 ms / 2,000 ms
コード長 1,438 bytes
コンパイル時間 2,032 ms
コンパイル使用メモリ 201,280 KB
最終ジャッジ日時 2025-01-23 07:48:06
ジャッジサーバーID
(参考情報)
judge3 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 19
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;
#define rep(i, n) for(int i = 0; i < n; i++)
#define rep2(i, x, n) for(int i = x; i <= n; i++)
#define rep3(i, x, n) for(int i = x; i >= n; i--)
#define each(e, v) for(auto &e: v)
#define pb push_back
#define eb emplace_back
#define all(x) x.begin(), x.end()
#define rall(x) x.rbegin(), x.rend()
#define sz(x) (int)x.size()
using ll = long long;
using pii = pair<int, int>;
using pil = pair<int, ll>;
using pli = pair<ll, int>;
using pll = pair<ll, ll>;
const int MOD = 1000000007;
//const int MOD = 998244353;
const int inf = (1<<30)-1;
const ll INF = (1LL<<60)-1;
template<typename T> bool chmax(T &x, const T &y) {return (x < y)? (x = y, true) : false;};
template<typename T> bool chmin(T &x, const T &y) {return (x > y)? (x = y, true) : false;};

struct io_setup{
    io_setup(){
        ios_base::sync_with_stdio(false);
        cin.tie(NULL);
        cout << fixed << setprecision(15);
    }
} io_setup;

int main(){
    int N; cin >> N;

    int M = 3000;
    vector<vector<int>> a(3, vector<int>(N));
    vector<vector<ll>> b(3, vector<ll>(M+1, 0));

    rep(i, 3){
        rep(j, N){
            cin >> a[i][j];
            b[i][a[i][j]]++;
        }
    }

    ll ans = 0;

    vector<ll> c(M+2, 0);
    rep(i, M+1) c[i+1] = c[i]+b[2][i];

    rep2(i, 1, M) rep2(j, 1, i){
        int L = i-j+1, R = i+1;
        ans += b[0][i]*b[1][j]*(c[R]-c[L]);
    }

    cout << ans << '\n';
}
0