답안 #440531

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
440531 2021-07-02T11:31:02 Z Abrar_Al_Samit Spiderman (COCI20_spiderman) C++17
70 / 70
1704 ms 126696 KB
#include<bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>

using namespace std;
using namespace __gnu_pbds;


#define debug(x) cerr << '[' << (#x) << "] = " << x << '\n';

template<class T> using ordered_set = tree<T, null_type , less<T> , rb_tree_tag , tree_order_statistics_node_update> ;

const int maxn = 1e6 + 6;
vector<int>divisor[maxn];
void sieve() {  
    for(int i=1; i<maxn; ++i) {
        for(int j=i; j<maxn; j+=i)  
            divisor[j].push_back(i); 
    }
}

void PlayGround() {
    int n; cin >> n;
    int k; cin >> k;
    vector<int>freq(maxn), ans(maxn);
    vector<int>h(n);
    for(int i=0; i<n; ++i) {
        cin >> h[i];
        freq[h[i]]++;
    }
    int seen = 0;
    for(int i=1; i<maxn; ++i) {
        seen += freq[i];
        if(freq[i]==0) continue;
        if(i==k) {
            ans[i] += n - seen;
        } else if(i > k) {
            int x = i - k;
            for(auto d : divisor[x]) 
                if(i % d == k) ans[i] += freq[d];
            if(x==i) --ans[i];
        }
    }
    for(int i=0; i<n; ++i) {
        cout << ans[h[i]] << ' ';
    }
    cout << '\n';

    #ifndef ONLINE_JUDGE
        cerr << "Time elapsed: " << 1.0 * clock() / CLOCKS_PER_SEC << " s.\n";
    #endif
} 
int main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    // #ifndef ONLINE_JUDGE
    //     freopen("input.txt", "r", stdin);
    // #endif
    sieve();
    PlayGround();

    return 0;
}
# 결과 실행 시간 메모리 Grader output
1 Correct 1587 ms 122720 KB Output is correct
2 Correct 1622 ms 122684 KB Output is correct
3 Correct 1649 ms 123976 KB Output is correct
4 Correct 1670 ms 126500 KB Output is correct
5 Correct 1650 ms 123948 KB Output is correct
6 Correct 1684 ms 126696 KB Output is correct
7 Correct 1658 ms 123976 KB Output is correct
8 Correct 1651 ms 123896 KB Output is correct
9 Correct 1703 ms 126512 KB Output is correct
10 Correct 1704 ms 126520 KB Output is correct