結果

問題 No.1623 三角形の制作
ユーザー ktr216
提出日時 2021-07-23 22:44:40
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,306 bytes
コンパイル時間 1,483 ms
コンパイル使用メモリ 173,784 KB
実行使用メモリ 145,152 KB
最終ジャッジ日時 2024-07-18 18:49:56
合計ジャッジ時間 8,221 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 1 WA * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#define rep(i, l, r) for (int i = (l); i < (r); i++)
using namespace std;

typedef long long ll;

int gcd(int x, int y) {
    if (y == 0) return x;
    return gcd(y, x % y);
}

int main() {
    int n, x;
    cin >> n;
    vector<int> r(n), g(3001, 0), b(3001, 0);
    rep(i, 0, n) {
        cin >> r[i];
    }
    rep(i, 0, n) {
        cin >> x;
        g[x]++;
    }
    rep(i, 0, n) {
        cin >> x;
        b[x]++;
    }
    vector<vector<ll>> a(3001, vector<ll>(3001, 0));
    rep(i, 0, 3000) {
        rep(j, 0, 3000) {
            a[i][j] = g[i] * b[j];
        }
    }
    vector<vector<ll>> s(3001, vector<ll>(3001, 0));
    rep(i, 0, 3000) {
        rep(j, 0, 3000) {
            s[i + 1][j + 1] = s[i][j + 1] + s[i + 1][j] - s[i][j] + a[i + 1][j + 1];
        }
    }
    vector<ll> d(6001, 0);
    rep(k, 0, 6001) {
        if (k > 0) d[k] = d[k - 1];
        rep(i, max(0, k - 3000), min(3000, k) + 1) {
            d[k] += a[i][k - i];
        }
    }
    /*
    rep(i, 0, 4) {
        rep(j, 0, 4) {
            cout << s[i + 1][j + 1];
        }
        cout << endl;
    }
    rep(k, 0, 6) cout << "d[" << k << "]=" << d[k] << endl;
    */
    ll ans = 0;
    rep(i, 0, n) {
        ans += s[r[i]][r[i]];
        ans -= d[r[i]];
    }
    cout << ans << endl;
}
0