結果

問題 No.871 かえるのうた
ユーザー Y17
提出日時 2019-08-30 22:57:15
言語 C++11(廃止可能性あり)
(gcc 13.3.0)
結果
AC  
実行時間 195 ms / 2,000 ms
コード長 830 bytes
コンパイル時間 1,366 ms
コンパイル使用メモリ 174,832 KB
実行使用メモリ 16,128 KB
最終ジャッジ日時 2024-11-30 10:49:42
合計ジャッジ時間 5,219 ms
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 49
権限があれば一括ダウンロードができます

ソースコード

diff #

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

#define int long long
signed main(){
    int n, k;
    cin >> n >> k;

    int x[100010];
    int a[100010];
    for(int i = 0;i < n;i++) cin >> x[i];
    for(int i = 0;i < n;i++) cin >> a[i];

    map<int, int> mp;
    set<int> st;

    for(int i = 0;i < n;i++){
        mp[x[i]] = i;
        if(i != k-1) st.insert(x[i]);
    }

    int ans = 0;

    queue<int> que;
    que.push(x[k-1]);

    while(!que.empty()){
        int idx = mp[que.front()];
        que.pop();
        ans++;

        int l = x[idx]-a[idx];
        int r = x[idx]+a[idx];

        for(auto itr = st.lower_bound(l);itr != st.end() && *itr <= r;){
            que.push(*itr);
            int tmp = *itr;
            *itr++;
            st.erase(tmp);
        }
    }

    cout << ans << endl;
    return 0;
}
0