結果

問題 No.1597 Matrix Sort
ユーザー ぷら
提出日時 2021-07-09 22:16:46
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 175 ms / 1,500 ms
コード長 1,056 bytes
コンパイル時間 2,025 ms
コンパイル使用メモリ 195,832 KB
最終ジャッジ日時 2025-01-22 22:01:15
ジャッジサーバーID
(参考情報)
judge3 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;

int main() {
    int N,P;
    long long K;
    cin >> N >> K >> P;
    vector<int>sum(P+1);
    vector<int>A(N);
    for(int i = 0; i < N; i++) {
        cin >> A[i];
        A[i] %= P;
    }
    for(int i = 0; i < N; i++) {
        int B;
        cin >> B;
        sum[B%P]++;
    }
    for(int i = 0; i < P; i++) {
        sum[i+1] += sum[i];
    }
    int l = -1,r = P-1;
    while (l+1 < r) {
        int mid = (l+r)/2;
        long long sum2 = 0;
        for(int i = 0; i < N; i++) {
            int L = (P-A[i])%P,R = (P+mid-A[i])%P;
            if(L <= R) {
                sum2 += sum[R];
                if(L) {
                    sum2 -= sum[L-1];
                }
            }
            else {
                sum2 += sum[P];
                if(L) {
                    sum2 -= sum[L-1];
                }
                sum2 += sum[R];
            }
        }
        if(sum2 < K) {
            l = mid;
        }
        else {
            r = mid;
        }
    }
    cout << r << endl;
}
0