Submission #119701

#	Time	Username	Problem	Language	Result	Execution time	Memory
119701		eriksuenderhauf	Pairs (IOI07_pairs)	C++11	60 / 100	42 ms	2812 KiB

pairs

//#pragma GCC optimize("O3")
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
#include <ext/rope>
#define mem(a,v) memset((a), (v), sizeof (a))
#define enl printf("\n")
#define case(t) printf("Case #%d: ", (t))
#define ni(n) scanf("%d", &(n))
#define nl(n) scanf("%lld", &(n))
#define nai(a, n) for (int i = 0; i < (n); i++) ni(a[i])
#define nal(a, n) for (int i = 0; i < (n); i++) nl(a[i])
#define pri(n) printf("%d\n", (n))
#define prl(n) printf("%lld\n", (n))
#define pii pair<int, int>
#define pil pair<int, long long>
#define pll pair<long long, long long>
#define vii vector<pii>
#define vil vector<pil>
#define vll vector<pll>
#define vi vector<int>
#define vl vector<long long>
#define pb push_back
#define mp make_pair
#define fi first
#define se second
using namespace std;
using namespace __gnu_pbds;
typedef long long ll;
typedef cc_hash_table<int,int,hash<int>> ht;
typedef tree<int,null_type,less<int>,rb_tree_tag,tree_order_statistics_node_update> oset;
const double pi = acos(-1);
const int MOD = 1e9 + 7;
const int INF = 1e9 + 7;
const int MAX2 = 263000 + 5;
const double eps = 1e-9;
int b, n, d, m;
int BIT2[MAX2];

void upd2(int ind, int v) {
	ind++;
	while (ind < MAX2) {
		BIT2[ind] += v;
		ind += ind & -ind;
	}
}

int qry2(int ind) {
	ind++;
	int ret = 0;
	while (ind > 0) {
		ret += BIT2[ind];
		ind -= ind & -ind;
	}
	return ret;
}

ll qry2(int l, int r) {
	return qry2(r) - qry2(l - 1);
}

int main() {
	scanf("%d %d %d %d", &b, &n, &d, &m);
	ll ans = 0;
	if (b == 1) {
		vector<int> a;
		a.resize(n);
		for (int i = 0; i < n; i++)
			ni(a[i]);
		sort(a.begin(), a.end());
		int j = 0;
		for (int i = 0; i < n; i++) {
			while (j <= i && a[i]-a[j] > d) j++;
			ans += (ll) (i - j);
		}
	} else if (b == 2) {
		vector<pii> a;
		a.resize(n);
		for (int i = 0; i < n; i++) {
			scanf("%d %d", &a[i].fi, &a[i].se);
			a[i] = mp(a[i].fi-a[i].se, a[i].fi+a[i].se);
		}
		sort(a.begin(), a.end());
		int j = 0; 
		for (int i = 0; i < n; i++) {
			while (j <= i && a[i].fi-a[j].fi > d)
				upd2(a[j++].se, -1);
			ans += qry2(a[i].se - d, a[i].se + d);
			upd2(a[i].se, 1);
		}
	} else {
		
	}
	prl(ans);
    return 0;
}

Compilation message (stderr)

pairs.cpp: In function 'int main()':
pairs.cpp:63:7: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
  scanf("%d %d %d %d", &b, &n, &d, &m);
  ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
pairs.cpp:9:20: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
 #define ni(n) scanf("%d", &(n))
               ~~~~~^~~~~~~~~~~~
pairs.cpp:69:4: note: in expansion of macro 'ni'
    ni(a[i]);
    ^~
pairs.cpp:80:9: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
    scanf("%d %d", &a[i].fi, &a[i].se);
    ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 6 / 6
• Subtask 2: 6 / 6
• Subtask 3: 6 / 6
• Subtask 4: 6 / 6
• Subtask 5: 6 / 6
• Subtask 6: 7 / 7
• Subtask 7: 7 / 7
• Subtask 8: 8 / 8
• Subtask 9: 8 / 8
• Subtask 10: 0 / 10
• Subtask 11: 0 / 10
• Subtask 12: 0 / 10
• Subtask 13: 0 / 10