Submission #1304223

#TimeUsernameProblemLanguageResultExecution timeMemory
1304223edgrigSpiderman (COCI20_spiderman)C++20
Compilation error
0 ms0 KiB
#define _CRT_SECURE_NO_WARNINGS
#include <iostream>
#include <algorithm>
#include <cmath>
#include <map>
#include <unordered_map>
#include <set>
#include <unordered_set>
#include <vector>
#include <array>
#include <list>
#include <queue>
#include <stack>
#include <iomanip>
#include <chrono>
#include <random>
#include <stdint.h>
#include <string>
#include <bitset>

using namespace std;

typedef long long ll;
typedef unsigned long long ull;

using pi = pair<int, int>;
using pll = pair<ll, ll>;
using vi = vector<int>;
using vll = vector<ll>;
using vpi = vector<pi>;
using vpll = vector<pll>;
using mii = map<int, int>;
using mll = map<ll, ll>;

#define ff first
#define ss second
//#define mp make_pair
#define pb push_back
#define eb emplace_back
#define pob pop_back
#define lb lower_bound
#define ub upper_bound
#define bs binary_search

const ll MOD = ll(1e9 + 7);
const ll MOD2 = ll(998244353);

mt19937 rnd(chrono::steady_clock::now().time_since_epoch().count());

void fastio() {
	ios_base::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	return;
}

ull gcd(ull a, ull b) {
	while (b) {
		a %= b;
		swap(a, b);
	}
	return a;
}

ull lcm(ull a, ull b) {
	return a / gcd(a, b) * b;
}

string dec_to_bin(ll a) {
	string s = "";
	for (ll i = a; i > 0; ) {
		ll k = i % 2;
		i /= 2;
		char c = k + 48;
		s += c;
	}
	if (a == 0) {
		s = "0";
	}
	reverse(s.begin(), s.end());
	return s;
}

ll bin_to_dec(string s) {
	ll num = 0;
	for (int i = 0; i < s.size(); i++) {
		num *= 2ll;
		num += (s[i] - '0');
	}
	return num;
}

bool isPrime(ll a) {
	if (a == 1) {
		return false;
	}
	for (ll i = 2; i * i <= a; i++) {
		if (a % i == 0) {
			return false;
		}
	}
	return true;
}

int dig_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;
}

string lex_small(string s, string t)
{
	if (s < t) return s;
	return t;
}

vector<int> find_divisors(int a)
{
	vector<int> ret;
	int sq = sqrt(a);
	for (int i = 1; i <= sq; i++)
	{
		if (a % i == 0)
		{
			ret.pb(i);
			if (i * i != a)
			{
				ret.pb(a / i);
			}
		}
	}
	return ret;
}

void solve()
{
	int n, k;
	cin >> n >> k;
	vector<int> h(n), h1(n);
	map<int, int> mp;
	vector<vector<int> > div(n);
	for (int i = 0; i < n; i++)
	{
		cin >> h[i];
		mp[h[i]]++;
	}
	h1 = h;
	sort(h1.begin(), h1.end());
	for (int i = 0; i < n; i++)
	{
		if (h[i] == k) div[i].pb(-2);
		else if (h[i] < k) div[i].pb(-1);
		else div[i] = find_divisors(h[i] - k);
	}
	vector<int> ans(n);
	for (int i = 0; i < n; i++)
	{
		if (div[i][0] == -1) ans[i] = 0;
		else if (div[i][0] == -2)
		{
			int ind = upper_bound(h1.begin(), h1.end(), h1[i]) - h1.begin();
			ans[i] = n - ind;
		}
		else
		{
			for (auto v : div[i])
			{
				if (v > k || !k)
					ans[i] += mp[v];
				if (v == h[i]) ans[i]--;
			}
		}
	}
	for (int i = 0; i < n; i++)
	{
		cout << ans[i] << " ";
	}
	cout << "\n";
}

int main()
{
	int test_cases = 1;
	//cin >> test_cases;

	while (test_cases--)
	{
		solve();
	}
	return 0;
}#define _CRT_SECURE_NO_WARNINGS
#include <iostream>
#include <algorithm>
#include <cmath>
#include <map>
#include <unordered_map>
#include <set>
#include <unordered_set>
#include <vector>
#include <array>
#include <list>
#include <queue>
#include <stack>
#include <iomanip>
#include <chrono>
#include <random>
#include <stdint.h>
#include <string>
#include <bitset>

using namespace std;

typedef long long ll;
typedef unsigned long long ull;

using pi = pair<int, int>;
using pll = pair<ll, ll>;
using vi = vector<int>;
using vll = vector<ll>;
using vpi = vector<pi>;
using vpll = vector<pll>;
using mii = map<int, int>;
using mll = map<ll, ll>;

#define ff first
#define ss second
//#define mp make_pair
#define pb push_back
#define eb emplace_back
#define pob pop_back
#define lb lower_bound
#define ub upper_bound
#define bs binary_search

const ll MOD = ll(1e9 + 7);
const ll MOD2 = ll(998244353);

mt19937 rnd(chrono::steady_clock::now().time_since_epoch().count());

void fastio() {
	ios_base::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	return;
}

ull gcd(ull a, ull b) {
	while (b) {
		a %= b;
		swap(a, b);
	}
	return a;
}

ull lcm(ull a, ull b) {
	return a / gcd(a, b) * b;
}

string dec_to_bin(ll a) {
	string s = "";
	for (ll i = a; i > 0; ) {
		ll k = i % 2;
		i /= 2;
		char c = k + 48;
		s += c;
	}
	if (a == 0) {
		s = "0";
	}
	reverse(s.begin(), s.end());
	return s;
}

ll bin_to_dec(string s) {
	ll num = 0;
	for (int i = 0; i < s.size(); i++) {
		num *= 2ll;
		num += (s[i] - '0');
	}
	return num;
}

bool isPrime(ll a) {
	if (a == 1) {
		return false;
	}
	for (ll i = 2; i * i <= a; i++) {
		if (a % i == 0) {
			return false;
		}
	}
	return true;
}

int dig_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;
}

string lex_small(string s, string t)
{
	if (s < t) return s;
	return t;
}

vector<int> find_divisors(int a)
{
	vector<int> ret;
	int sq = sqrt(a);
	for (int i = 1; i <= sq; i++)
	{
		if (a % i == 0)
		{
			ret.pb(i);
			if (i * i != a)
			{
				ret.pb(a / i);
			}
		}
	}
	return ret;
}

void solve()
{
	int n, k;
	cin >> n >> k;
	vector<int> h(n), h1(n);
	map<int, int> mp;
	vector<vector<int> > div(n);
	for (int i = 0; i < n; i++)
	{
		cin >> h[i];
		mp[h[i]]++;
	}
	h1 = h;
	sort(h1.begin(), h1.end());
	for (int i = 0; i < n; i++)
	{
		if (h[i] == k) div[i].pb(-2);
		else if (h[i] < k) div[i].pb(-1);
		else div[i] = find_divisors(h[i] - k);
	}
	vector<int> ans(n);
	for (int i = 0; i < n; i++)
	{
		if (div[i][0] == -1) ans[i] = 0;
		else if (div[i][0] == -2)
		{
			int ind = upper_bound(h1.begin(), h1.end(), h1[i]) - h1.begin();
			ans[i] = n - ind;
		}
		else
		{
			for (auto v : div[i])
			{
				if (v > k || !k)
					ans[i] += mp[v];
				if (v == h[i]) ans[i]--;
			}
		}
	}
	for (int i = 0; i < n; i++)
	{
		cout << ans[i] << " ";
	}
	cout << "\n";
}

int main()
{
	int test_cases = 1;
	//cin >> test_cases;

	while (test_cases--)
	{
		solve();
	}
	return 0;
}

Compilation message (stderr)

spiderman.cpp:237:2: error: stray '#' in program
  237 | }#define _CRT_SECURE_NO_WARNINGS
      |  ^
spiderman.cpp:237:3: error: 'define' does not name a type
  237 | }#define _CRT_SECURE_NO_WARNINGS
      |   ^~~~~~
spiderman.cpp:281:10: error: redefinition of 'const ll MOD'
  281 | const ll MOD = ll(1e9 + 7);
      |          ^~~
spiderman.cpp:45:10: note: 'const ll MOD' previously defined here
   45 | const ll MOD = ll(1e9 + 7);
      |          ^~~
spiderman.cpp:282:10: error: redefinition of 'const ll MOD2'
  282 | const ll MOD2 = ll(998244353);
      |          ^~~~
spiderman.cpp:46:10: note: 'const ll MOD2' previously defined here
   46 | const ll MOD2 = ll(998244353);
      |          ^~~~
spiderman.cpp:284:9: error: redefinition of 'std::mt19937 rnd'
  284 | mt19937 rnd(chrono::steady_clock::now().time_since_epoch().count());
      |         ^~~
spiderman.cpp:48:9: note: 'std::mt19937 rnd' previously declared here
   48 | mt19937 rnd(chrono::steady_clock::now().time_since_epoch().count());
      |         ^~~
spiderman.cpp:286:6: error: redefinition of 'void fastio()'
  286 | void fastio() {
      |      ^~~~~~
spiderman.cpp:50:6: note: 'void fastio()' previously defined here
   50 | void fastio() {
      |      ^~~~~~
spiderman.cpp:293:5: error: redefinition of 'ull gcd(ull, ull)'
  293 | ull gcd(ull a, ull b) {
      |     ^~~
spiderman.cpp:57:5: note: 'ull gcd(ull, ull)' previously defined here
   57 | ull gcd(ull a, ull b) {
      |     ^~~
spiderman.cpp:301:5: error: redefinition of 'ull lcm(ull, ull)'
  301 | ull lcm(ull a, ull b) {
      |     ^~~
spiderman.cpp:65:5: note: 'ull lcm(ull, ull)' previously defined here
   65 | ull lcm(ull a, ull b) {
      |     ^~~
spiderman.cpp:305:8: error: redefinition of 'std::string dec_to_bin(ll)'
  305 | string dec_to_bin(ll a) {
      |        ^~~~~~~~~~
spiderman.cpp:69:8: note: 'std::string dec_to_bin(ll)' previously defined here
   69 | string dec_to_bin(ll a) {
      |        ^~~~~~~~~~
spiderman.cpp:320:4: error: redefinition of 'll bin_to_dec(std::string)'
  320 | ll bin_to_dec(string s) {
      |    ^~~~~~~~~~
spiderman.cpp:84:4: note: 'll bin_to_dec(std::string)' previously defined here
   84 | ll bin_to_dec(string s) {
      |    ^~~~~~~~~~
spiderman.cpp:329:6: error: redefinition of 'bool isPrime(ll)'
  329 | bool isPrime(ll a) {
      |      ^~~~~~~
spiderman.cpp:93:6: note: 'bool isPrime(ll)' previously defined here
   93 | bool isPrime(ll a) {
      |      ^~~~~~~
spiderman.cpp:341:5: error: redefinition of 'int dig_sum(ll)'
  341 | int dig_sum(ll a) {
      |     ^~~~~~~
spiderman.cpp:105:5: note: 'int dig_sum(ll)' previously defined here
  105 | int dig_sum(ll a) {
      |     ^~~~~~~
spiderman.cpp:350:4: error: redefinition of 'll binpow(ll, int)'
  350 | ll binpow(ll a, int b) {
      |    ^~~~~~
spiderman.cpp:114:4: note: 'll binpow(ll, int)' previously defined here
  114 | ll binpow(ll a, int b) {
      |    ^~~~~~
spiderman.cpp:362:4: error: redefinition of 'll binpow_by_mod(ll, ll, ll)'
  362 | ll binpow_by_mod(ll a, ll b, ll mod) {
      |    ^~~~~~~~~~~~~
spiderman.cpp:126:4: note: 'll binpow_by_mod(ll, ll, ll)' previously defined here
  126 | ll binpow_by_mod(ll a, ll b, ll mod) {
      |    ^~~~~~~~~~~~~
spiderman.cpp:376:4: error: redefinition of 'll factorial(int)'
  376 | ll factorial(int a) {
      |    ^~~~~~~~~
spiderman.cpp:140:4: note: 'll factorial(int)' previously defined here
  140 | ll factorial(int a) {
      |    ^~~~~~~~~
spiderman.cpp:384:4: error: redefinition of 'll factorial_by_mod(int, ll)'
  384 | ll factorial_by_mod(int a, ll mod) {
      |    ^~~~~~~~~~~~~~~~
spiderman.cpp:148:4: note: 'll factorial_by_mod(int, ll)' previously defined here
  148 | ll factorial_by_mod(int a, ll mod) {
      |    ^~~~~~~~~~~~~~~~
spiderman.cpp:393:8: error: redefinition of 'std::string lex_small(std::string, std::string)'
  393 | string lex_small(string s, string t)
      |        ^~~~~~~~~
spiderman.cpp:157:8: note: 'std::string lex_small(std::string, std::string)' previously defined here
  157 | string lex_small(string s, string t)
      |        ^~~~~~~~~
spiderman.cpp:399:13: error: redefinition of 'std::vector<int> find_divisors(int)'
  399 | vector<int> find_divisors(int a)
      |             ^~~~~~~~~~~~~
spiderman.cpp:163:13: note: 'std::vector<int> find_divisors(int)' previously defined here
  163 | vector<int> find_divisors(int a)
      |             ^~~~~~~~~~~~~
spiderman.cpp:417:6: error: redefinition of 'void solve()'
  417 | void solve()
      |      ^~~~~
spiderman.cpp:181:6: note: 'void solve()' previously defined here
  181 | void solve()
      |      ^~~~~
spiderman.cpp:463:5: error: redefinition of 'int main()'
  463 | int main()
      |     ^~~~
spiderman.cpp:227:5: note: 'int main()' previously defined here
  227 | int main()
      |     ^~~~