# |
Author |
Problem |
Lang |
Verdict |
Time |
Memory |
Sent |
Judged |
|
207676206 |
Practice:
Tsovak |
1830B
- 32
|
C++20 (GCC 11-64)
|
Accepted
|
733 ms
|
8452 KB
|
2023-05-28 21:28:26 |
2023-05-28 21:28:26 |
|
// -------------------- Includes -------------------- //
#define _CRT_SECURE_NO_WARNINGS
#define _USE_MATH_DEFINES
#include <iostream>
#include <iomanip>
#include <cstdio>
#include <stdio.h>
#include <cstdlib>
#include <stdlib.h>
#include <array>
#include <string>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <math.h>
#include <set>
#include <map>
#include <unordered_set>
#include <unordered_map>
#include <vector>
#include <stack>
#include <queue>
#include <deque>
#include <bitset>
#include <list>
#include <iterator>
#include <numeric>
#include <complex>
#include <tuple>
#include <utility>
#include <cassert>
#include <assert.h>
#include <climits>
#include <limits.h>
#include <ctime>
#include <time.h>
#include <random>
#include <chrono>
#include <fstream>
//#include <decimal>
using namespace std;
// -------------------- Typedefs -------------------- //
typedef long long ll;
typedef unsigned long long ull;
typedef double db;
typedef long double ld;
// -------------------- Defines -------------------- //
#define pr pair
#define mpr make_pair
#define ff first
#define ss second
#define mset multiset
#define mmap multimap
#define uset unordered_set
#define umap unordered_map
#define umset unordered_multiset
#define ummap unordered_multimap
#define pqueue priority_queue
#define sz(x) (int((x).size()))
#define len(x) (int((x).length()))
#define all(x) (x).begin(), (x).end()
#define rall(x) (x).rbegin(), (x).rend()
#define clr(x) (x).clear()
#define pf push_front
#define pb push_back
#define popf pop_front
#define popb pop_back
#define ft front
#define bk back
#define lb lower_bound
#define ub upper_bound
#define bs binary_search
// -------------------- Constants -------------------- //
const int MAX = int(1e9 + 5);
const ll MAXL = ll(1e18 + 5);
const ll MOD = ll(1e9 + 7);
const ll MOD2 = ll(998244353);
// -------------------- Functions -------------------- //
void fastio()
{
ios_base::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
return;
}
void precision(int x)
{
cout.setf(ios::fixed | ios::showpoint);
cout.precision(x);
return;
}
ll gcd(ll a, ll b)
{
while (b) {
a %= b;
swap(a, b);
}
return a;
}
ll lcm(ll a, ll b)
{
return a / gcd(a, b) * b;
}
bool is_prime(ll a)
{
if (a == 1) {
return false;
}
for (ll i = 2; i * i <= a; i++) {
if (a % i == 0) {
return false;
}
}
return true;
}
bool is_square(ll a)
{
ll b = ll(sqrt(a));
return (b * b == a);
}
bool is_cube(ll a)
{
ll b = ll(cbrt(a));
return (b * b * b == a);
}
int digit_sum(ll a)
{
int sum = 0;
while (a) {
sum += int(a % 10);
a /= 10;
}
return sum;
}
ll binpow(ll a, int b)
{
ll ans = 1;
while (b) {
if ((b & 1) == 1) {
ans *= a;
}
b >>= 1;
a *= a;
}
return ans;
}
ll binpow_by_mod(ll a, ll b, ll mod)
{
ll ans = 1;
while (b) {
if ((b & 1) == 1) {
ans *= a;
ans %= mod;
}
b >>= 1;
a *= a;
a %= mod;
}
return ans;
}
ll factorial(int a)
{
ll ans = 1;
for (int i = 2; i <= a; i++) {
ans *= ll(i);
}
return ans;
}
ll factorial_by_mod(int a, ll mod)
{
ll ans = 1;
for (int i = 2; i <= a; i++) {
ans *= ll(i);
ans %= mod;
}
return ans;
}
// -------------------- Solution -------------------- //
/*const int N = 200005;
int a[N];
int n;*/
void solve()
{
int i, j;
int n;
cin >> n;
vector<int> a(n), b(n);
vector<vector<int>> c(n + 1);
for (i = 0; i < n; i++) {
cin >> a[i];
}
for (i = 0; i < n; i++) {
cin >> b[i];
c[a[i]].pb(b[i]);
}
for (i = 1; i <= n; i++) sort(all(c[i]));
ll ans = 0;
int k, l, r;
bool ok;
for (i = 1; i * (i + 1) <= 2 * n && i <= n; i++) {
for (j = i + 1; i * j <= 2 * n && j <= n; j++) {
if (c[i].empty() || c[j].empty()) continue;
ok = false;
if (sz(c[i]) > sz(c[j])) {
swap(i, j);
ok = true;
}
for (k = 0; k < sz(c[i]); k++) {
if (c[i][k] >= i * j) break;
l = lb(all(c[j]), i * j - c[i][k]) - c[j].begin();
if (l == sz(c[j]) || c[j][l] != i * j - c[i][k]) continue;
r = ub(all(c[j]), i * j - c[i][k]) - c[j].begin();
ans += ll(r - l);
}
if (ok) swap(i, j);
}
}
for (i = 1; i * i <= 2 * n; i++) {
for (j = 0; j < sz(c[i]); j++) {
if (2 * c[i][j] >= i * i) break;
l = lb(all(c[i]), i * i - c[i][j]) - c[i].begin();
if (l == sz(c[i]) || c[i][l] != i * i - c[i][j]) continue;
r = ub(all(c[i]), i * i - c[i][j]) - c[i].begin();
ans += ll(r - l);
}
if (i % 2 == 0) {
k = i * i / 2;
l = lb(all(c[i]), k) - c[i].begin();
if (l == sz(c[i]) || c[i][l] != k) continue;
r = ub(all(c[i]), k) - c[i].begin();
j = r - l;
ans += (ll(j) * ll(j - 1) / 2LL);
}
}
cout << ans << "\n";
return;
}
void precalc()
{
return;
}
int main()
{
fastio();
precalc();
int tests = 1, tc;
cin >> tests;
for (tc = 1; tc <= tests; tc++) {
solve();
}
//cout << db(clock()) / CLOCKS_PER_SEC << endl;
return 0;
}
/*
# # # # # # # # # # # # # # # # #
# # # # # # # # # #
# # # # # # # # # # # # #
# # # # # # # # # # # #
# # # # # # # # # # # # # #
*/
Click to see test details