제출 #547238

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
547238		Soul234	Pairs (IOI07_pairs)	C++14	100 / 100	298 ms	33356 KiB

pairs

#include<bits/stdc++.h>
using namespace std;

void DBG() { cerr << "]\n"; }
template<class H, class... T> void DBG(H h, T... t) {
    cerr << h; if(sizeof...(t)) cerr << ", ";
    DBG(t...);
}
#ifdef LOCAL
#define dbg(...) cerr << "[" << #__VA_ARGS__ << "]: [", DBG(__VA_ARGS__)
#else
#define dbg(...) 0
#endif // LOCAL

#define FOR(i,a,b) for(int i = (a) ; i<(b) ; i++)
#define F0R(i,a) FOR(i,0,a)
#define ROF(i,a,b) for(int i = (b)-1 ; i>=(a) ; i--)
#define R0F(i,a) ROF(i,0,a)
#define each(e,a) for(auto &e : (a))
#define sz(v) (int)(v).size()
#define all(v) (v).begin(),(v).end()
#define rall(v) (v).rbegin(),(v).rend()
#define pb push_back
#define tcT template<class T
#define nl "\n"

using ll = long long;
using vi = vector<int>;
using pi = pair<int,int>;
using str = string;
tcT> using V = vector<T>;
tcT> using pqg = priority_queue<T,vector<T>,greater<T>>;

void setIO(string NAME = "") {
    cin.tie(0)->sync_with_stdio(0);
    if(sz(NAME)) {
        freopen((NAME + ".inp").c_str(),"r",stdin);
        freopen((NAME + ".out").c_str(),"w",stdout);
    }
}

tcT, int SZ> using AR = array<T,SZ>;

tcT, int... Ns> struct BIT {
    T val = 0; void upd(T v) { val += v; }
    T query() { return val; }
};

tcT, int N, int... Ns> struct BIT<T, N, Ns...> {
    BIT<T, Ns...> bit[N+1];
    template<typename... Args> void upd(int pos, Args... args) {
        assert(pos>0);
        for(;pos<=N;pos+=pos&-pos) bit[pos].upd(args...);
    }
    template<typename... Args> T sum(int r, Args... args) {
        T res = 0; for(;r;r-=r&-r) res += bit[r].query(args...);
        return res;
    }
    template<typename... Args> T query(int l, int r, Args... args) {
        return sum(r, args...) - sum(l-1, args...);
    }
};

int B, N, D, M;
BIT<int, 305, 305, 305> bit1;

void solve() {
    cin>>B>>N>>D>>M;
    ll ans = 0;
    if(B == 1) {
        vi A(N);
        F0R(i,N) cin>>A[i];
        sort(all(A));
        int lo = 0, hi = 0;
        F0R(i,N) {
            while(hi+1<N && A[hi+1]-A[i] <= D) hi++;
            while(lo<N && A[i]-A[lo] > D) lo++;
            ans += hi-lo;
        }
    }
    else if(B == 2) {
        BIT<int, 150005> bit;
        V<AR<int,2>> A(N);
        F0R(i,N) {
            int x, y;
            cin>>x>>y;
            A[i] = {x+y, x-y};
        }
        sort(all(A));
        int lo = 0, hi = -1;
        F0R(i,N) {
            while(hi+1<N && A[hi+1][0] - A[i][0] <= D) bit.upd(A[++hi][1]+M+1, 1);
            while(lo<N && A[i][0] - A[lo][0] > D) bit.upd(A[lo++][1]+M+1, -1);
            ans += bit.query(max(1, A[i][1]-D+M+1), min(2*M+1, A[i][1]+D+M+1))-1;
        }
    }
    else if(B == 3) {
        V<AR<int,4>> A(N);
        F0R(i,N) {
            int x, y, z;
            cin>>x>>y>>z;
            A[i] = {x+y+z, x+y-z, x-y+z, x-y-z};
        }
        sort(all(A));
        int lo = 0, hi = -1;
        F0R(i,N) {
            while(hi+1<N && A[hi+1][0] - A[i][0] <= D) {
                hi++;
                bit1.upd(A[hi][1]+2*M+1, A[hi][2]+2*M+1, A[hi][3]+2*M+1, 1);
            }
            while(lo<N && A[i][0] - A[lo][0] > D) {
                bit1.upd(A[lo][1]+2*M+1, A[lo][2]+2*M+1, A[lo][3]+2*M+1, -1);
                lo++;
            }
            ans += bit1.query(max(1, A[i][1]-D+2*M+1), min(4*M+1, A[i][1]+D+2*M+1), max(1, A[i][2]-D+2*M+1), min(4*M+1, A[i][2]+D+2*M+1), max(1, A[i][3]-D+2*M+1), min(4*M+1, A[i][3]+D+2*M+1))-1;
        }
    }
    cout << ans/2 << nl;
}

int main() {
    setIO();

    int t=1;
    //cin>>t;
    while(t-->0) {
        solve();
    }

    return 0;
}

컴파일 시 표준 에러 (stderr) 메시지

pairs.cpp: In function 'void setIO(std::string)':
pairs.cpp:37:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   37 |         freopen((NAME + ".inp").c_str(),"r",stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
pairs.cpp:38:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   38 |         freopen((NAME + ".out").c_str(),"w",stdout);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 6 / 6
• Subtask 2: 6 / 6
• Subtask 3: 6 / 6
• Subtask 4: 6 / 6
• Subtask 5: 6 / 6
• Subtask 6: 7 / 7
• Subtask 7: 7 / 7
• Subtask 8: 8 / 8
• Subtask 9: 8 / 8
• Subtask 10: 10 / 10
• Subtask 11: 10 / 10
• Subtask 12: 10 / 10
• Subtask 13: 10 / 10