結果

問題 No.1597 Matrix Sort
コンテスト
ユーザー irona
提出日時 2021-07-10 11:15:21
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 165 ms / 1,500 ms
コード長 1,153 bytes
コンパイル時間 3,881 ms
コンパイル使用メモリ 231,464 KB
実行使用メモリ 83,072 KB
最終ジャッジ日時 2024-07-02 01:50:46
合計ジャッジ時間 8,037 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
#include <atcoder/all>
#define ll long long int
#define vec vector<ll>
#define mat vector<vector<ll>>
#define pll pair<ll,ll>
#define count __builtin_popcountll

using namespace std;
using namespace atcoder;
//using mint = modint1000000007;

const ll mod=1000000007;//998244353;
const ll inf=1000000000000000000;
ll dx4[4]={1,0,-1,0};
ll dy4[4]={0,-1,0,1};
ll dx8[8]={1,0,-1,1,-1,1,0,-1};
ll dy8[8]={1,1,1,0,0,-1,-1,-1};

int main(){
  cout << fixed << setprecision(15);

  ll n,k,p;
  cin >> n >> k >> p;
  vec a(n),b(n);
  for(ll i=0;i<n;i++)cin >> a[i];
  for(ll i=0;i<n;i++)cin >> b[i];
  for(ll i=0;i<n;i++){
    a[i]%=p;
    b[i]%=p;
  }
  


  sort(a.begin(),a.end());
  sort(b.begin(),b.end());


  vec s(p+1);
  for(ll i=0;i<n;i++){
    s[b[i]]++;
  }
  for(ll i=0;i<p-1;i++){
    s[i+1]+=s[i];
  }

  ll l=-1,r=p,h=(r+l)/2;

  while(abs(r-l)>1){
    ll ret=0;
    for(ll i=0;i<n;i++){
      if(a[i]<=h){
        ret+=s[h-a[i]];
        ret+=n-s[p-a[i]-1];
      }
      else{
        ret+=s[h-a[i]+p]-s[p-a[i]-1];
      }
    }
    if(ret>=k)r=h;
    else l=h;
    h=(r+l)/2;
  }

  cout << r << endl;
  

  return 0;
}
0