Submission #644928

#	Time	Username	Problem	Language	Result	Execution time	Memory
644928		1zaid1	Spiderman (COCI20_spiderman)	C++17	70 / 70	541 ms	19180 KiB

spiderman

#include <bits/stdc++.h>
using namespace std;
#define int long long
#define endl '\n';

const int M = 2e6+5, MOD = 1e9+7;
int fr[M], ans[M];

signed main() {
    cin.tie(0)->sync_with_stdio(0);

    int n, k;
    cin >> n >> k;

    vector<int> v(n);
    for (int&i:v) cin >> i;
    for (int i:v) fr[i]++;
    for (int&i:v) i -= k;

    for (int i:v) {
        if (i < 0) continue;
        if (ans[i]) continue; fr[i+k]--;
        for (int x = 1; x*x <= i; x++) {
            if (i%x == 0) {
                if ((i+k)%x == k) ans[i] += fr[x];
                if (x*x != i) if ((i+k)%(i/x) == k) ans[i] += fr[i/x];
            }
        } fr[i+k]++;
    }

    for (int i:v) ans[0] += i > 0;

    for (int i:v) {
        if (i >= 0) cout << ans[i] << ' ';
        else cout << 0 << ' ';
    } cout << endl;

    return 0;
}
/*
a%b = k
6 3
4 3 12 6 8 2
   0 1 2 3 4 5 6 7 8 9 10 11 12
fr:1 1 0 1 0 0 0 0 0 1 0  0  0
as:0 0 0 0 0 0 0 0 0 0 0  0  0
*/

Compilation message (stderr)

spiderman.cpp: In function 'int main()':
spiderman.cpp:22:9: warning: this 'if' clause does not guard... [-Wmisleading-indentation]
   22 |         if (ans[i]) continue; fr[i+k]--;
      |         ^~
spiderman.cpp:22:31: note: ...this statement, but the latter is misleadingly indented as if it were guarded by the 'if'
   22 |         if (ans[i]) continue; fr[i+k]--;
      |                               ^~

• Subtask 1: 70 / 70