QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#441618#8177. Sum is IntegerHKOI0#WA 199ms16812kbC++141.3kb2024-06-14 16:43:012024-06-14 16:43:03

Judging History

This is the latest submission verdict.

  • [2024-06-14 16:43:03]
  • Judged
  • Verdict: WA
  • Time: 199ms
  • Memory: 16812kb
  • [2024-06-14 16:43:01]
  • Submitted

answer

#include <bits/stdc++.h>
#define int long long
#define all(a) begin(a), end(a)
using namespace std;

const int N = 2e5 + 11;
int p[N], q[N];
long double r[N], s_r[N];
int s1[N], s2[N];

int pm(int a, int b, int p){
	if (b == 0) return 1;
	return pm(a * a % p, b / 2, p) * (b % 2 ? a : 1) % p;
}

int mi(int a, int p){
	return pm(a, p - 2, p);
}

void calc(int n, int mod, int* s){
	s[0] = 0;
	for (int i = 0; i < n; i++) {
		s[i + 1] = ((s[i] + p[i] * mi(q[i], mod)) % mod + mod) % mod; 
	}
	for (int i = 1; i <= n; i++) {
		s[i] = (s[i] - (int) floorl(s_r[i]) + mod) % mod;
	}
}

void solve() {
	int n; cin >> n;
	for (int i = 0; i < n; i++) {
		cin >> p[i] >> q[i]; r[i] = (long double) p[i] / q[i], s_r[i + 1] = s_r[i] + r[i];
	}

	const int MOD1 = 1e9 + 7;
	const int MOD2 = 998244353;
	calc(n, MOD1, s1);
	calc(n, MOD2, s2);

	map<pair<int, int>, int> mp;
	int ans = 0;
	for (int i = 0; i <= n; i++) {
		pair<int, int> p = {s1[i], s2[i]};
		for (int x = -1; x <= 1; x++) {
			for (int y = -1; y <= 1; y++) {
				ans += mp[{(p.first - x + MOD1) % MOD1, (p.second - y + MOD2) % MOD2}];
			}
		}
		mp[p]++;
	}
	cout << ans << '\n';
}

int32_t main(){
#ifndef LOCAL
	cin.tie(0)->sync_with_stdio(false);
#endif
	int t = 1;
	// cin >> t;
	while (t--) {
		solve();
	}
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 2ms
memory: 11712kb

input:

4
1 6
1 3
1 2
1 2

output:

2

result:

ok "2"

Test #2:

score: 0
Accepted
time: 0ms
memory: 9792kb

input:

5
1 1
2 2
3 3
4 4
5 5

output:

15

result:

ok "15"

Test #3:

score: 0
Accepted
time: 0ms
memory: 9616kb

input:

2
1 99999
99999 100000

output:

0

result:

ok "0"

Test #4:

score: -100
Wrong Answer
time: 199ms
memory: 16812kb

input:

200000
82781 82781
86223 86223
16528 16528
84056 84056
94249 94249
54553 54553
25943 25943
10415 10415
52417 52417
46641 46641
70298 70298
84228 84228
55441 55441
49326 49326
11753 11753
89499 89499
58220 58220
71482 71482
32373 32373
7251 7251
78573 78573
74268 74268
46682 46682
20314 20314
85519 8...

output:

19799405515

result:

wrong answer 1st words differ - expected: '10603308211', found: '19799405515'