結果

問題 No.728 ギブ and テイク
ユーザー kopricky
提出日時 2017-10-07 00:59:26
言語 C++11
(gcc 4.8.5)
結果
AC  
実行時間 182 ms
コード長 2,139 Byte
コンパイル時間 1,355 ms
使用メモリ 10,720 KB
最終ジャッジ日時 2019-02-21 09:32:01

テストケース

テストケース表示
入力 結果 実行時間
使用メモリ
00_sample1.txt AC 3 ms
6,872 KB
00_sample2.txt AC 2 ms
6,872 KB
00_sample3.txt AC 3 ms
6,872 KB
0_small_case1.txt AC 3 ms
6,868 KB
0_small_case2.txt AC 3 ms
8,916 KB
0_small_case3.txt AC 2 ms
6,872 KB
0_small_case4.txt AC 2 ms
6,868 KB
0_small_case5.txt AC 2 ms
8,916 KB
0_small_case6.txt AC 3 ms
6,872 KB
0_small_case7.txt AC 2 ms
6,872 KB
0_small_case8.txt AC 3 ms
6,872 KB
0_small_case9.txt AC 3 ms
6,868 KB
0_small_case10.txt AC 2 ms
6,872 KB
1_middle_case1.txt AC 13 ms
6,872 KB
1_middle_case2.txt AC 19 ms
6,872 KB
1_middle_case3.txt AC 9 ms
6,872 KB
1_middle_case4.txt AC 17 ms
6,872 KB
1_middle_case5.txt AC 17 ms
6,872 KB
2_large_case1.txt AC 130 ms
8,084 KB
2_large_case2.txt AC 140 ms
8,412 KB
2_large_case3.txt AC 163 ms
9,664 KB
2_large_case4.txt AC 147 ms
8,876 KB
2_large_case5.txt AC 60 ms
6,868 KB
2_large_case6.txt AC 39 ms
6,868 KB
2_large_case7.txt AC 107 ms
6,868 KB
2_large_case8.txt AC 105 ms
6,872 KB
2_large_case9.txt AC 56 ms
6,868 KB
2_large_case10.txt AC 182 ms
10,652 KB
3_challenge_case1.txt AC 123 ms
10,720 KB
3_challenge_case2.txt AC 2 ms
6,872 KB
3_challenge_case3.txt AC 2 ms
6,868 KB
3_challenge_case4.txt AC 2 ms
6,872 KB
3_challenge_case5.txt AC 2 ms
6,872 KB
テストケース一括ダウンロード

ソースコード

diff #
#include <bits/stdc++.h>
#define ll long long
#define INF 1000000005
#define MOD 1000000007
#define EPS 1e-10
#define rep(i,n) for(int i=0;i<(int)n;++i)
#define each(a,b) for(auto (a): (b))
#define all(v) (v).begin(),(v).end()
#define zip(v) sort(all(v)),v.erase(unique(all(v)),v.end())
#define fi first
#define se second
#define pb push_back
#define show(x) cout<<#x<<" = "<<(x)<<endl
#define spair(p) cout<<#p<<": "<<p.fi<<" "<<p.se<<endl
#define svec(v) cout<<#v<<":";rep(kbrni,v.size())cout<<" "<<v[kbrni];cout<<endl
#define sset(s) cout<<#s<<":";each(kbrni,s)cout<<" "<<kbrni;cout<<endl
#define smap(m) cout<<#m<<":";each(kbrni,m)cout<<" {"<<kbrni.first<<":"<<kbrni.second<<"}";cout<<endl

using namespace std;

typedef pair<int,int>P;

template<class V> class BIT {
private:
	int n; vector<V> bit;
public:
	void add(int i,V x){ i++; while(i < n) bit[i] += x, i += i & -i;}
	V sum(int i){i++; V s = 0; while(i>0) s += bit[i], i -= i & -i; return s;}
	BIT(int sz){ n = sz + 1, bit.resize(n,0);} //初期値がすべて0の場合
	BIT(vector<V> v){n = (int)v.size()+1; bit.resize(n); rep(i,n) add(i,v[i]);}
	void print(){ rep(i,n-1)cout<<sum(i)-sum(i-1)<< " ";cout<<endl;}
	void print_sum(){ rep(i,n)cout<<sum(i-1)<<" ";cout<<endl;}	//-1スタート
};

int main()
{
    int n;
    scanf("%d",&n);
	// assert(0 <= n && n <= 300000);
    vector<int> pos(n);
    vector<P> vec(2*n);
    rep(i,n){
        scanf("%d",&pos[i]);
		// assert(0 <= pos[i] && pos[i] <= 1000000000);
		// if(i > 0){
		// 	assert(pos[i-1] < pos[i]);
		// }
        vec[i] = P(pos[i],n+i);
    }
    vector<int> l(n),r(n);
    rep(i,n){
        scanf("%d%d",&l[i],&r[i]);
		// assert(0 <= l[i] && l[i] <= 1000000000 && 0 <= r[i] && r[i] <= 1000000000);
        vec[n+i] = P(pos[i]-l[i],i);
    }
    sort(all(vec));
    BIT<int> bt(n+5);
    ll ans = 0;
    rep(i,2*n){
        int p = vec[i].fi, q = vec[i].se;
        if(q >= n){
            q -= n;
            int id = upper_bound(all(pos),pos[q]+r[q]) - pos.begin();
            ans += bt.sum(id-1) - bt.sum(q);
        }else{
            bt.add(q,1);
        }
    }
    cout << ans << endl;
    return 0;
}
0