oj.uz

답안 #416894

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
416894 2021-06-03T07:02:00 Z codebuster_10 Pairs (IOI07_pairs) C++17
30 / 100
 33 ms 8812 KB 
#include <bits/stdc++.h>

using namespace std ;

#define int int64_t //be careful about this 
#define endl "\n"
#define f(i,a,b) for(int i=int(a);i<int(b);++i)

#define pr pair
#define ar array
#define fr first
#define sc second
#define vt vector
#define pb push_back
#define eb emplace_back
#define LB lower_bound  
#define UB upper_bound
#define PQ priority_queue

#define sz(x) ((int)(x).size())
#define all(a) (a).begin(),(a).end()
#define allr(a) (a).rbegin(),(a).rend()
#define mem0(a) memset(a, 0, sizeof(a))
#define mem1(a) memset(a, -1, sizeof(a))

template<class A> void rd(vt<A>& v);
template<class T> void rd(T& x){ cin >> x; }
template<class H, class... T> void rd(H& h, T&... t) { rd(h) ; rd(t...) ;}
template<class A> void rd(vt<A>& x) { for(auto& a : x) rd(a) ;}

template<class T> bool ckmin(T& a, const T& b) { return b < a ? a = b, 1 : 0; }
template<class T> bool ckmax(T& a, const T& b) { return a < b ? a = b, 1 : 0; }

template<typename T>
void __p(T a) {
  cout<<a; 
}
template<typename T, typename F>
void __p(pair<T, F> a) {
  cout<<"{";
  __p(a.first);
  cout<<",";
  __p(a.second);
  cout<<"}\n"; 
}
template<typename T>
void __p(std::vector<T> a) {
  cout<<"{";
  for(auto it=a.begin(); it<a.end(); it++)
    __p(*it),cout<<",}\n"[it+1==a.end()]; 
}
template<typename T, typename ...Arg>
void __p(T a1, Arg ...a) {
  __p(a1);
  __p(a...);
}
template<typename Arg1>
void __f(const char *name, Arg1 &&arg1) {
  cout<<name<<" : ";
  __p(arg1);
  cout<<endl;
}
template<typename Arg1, typename ... Args>
void __f(const char *names, Arg1 &&arg1, Args &&... args) {
  int bracket=0,i=0;
  for(;; i++)
    if(names[i]==','&&bracket==0)
      break;
    else if(names[i]=='(')
      bracket++;
    else if(names[i]==')')
      bracket--;
  const char *comma=names+i;
  cout.write(names,comma-names)<<" : ";
  __p(arg1);
  cout<<" | ";
  __f(comma+1,args...);
}

void setIO(string s = "") {
  ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); 
  cin.exceptions(cin.failbit); 
	cout.precision(15);	cout << fixed;
  #ifdef ONLINE_JUDGE
  if(sz(s)){
  	freopen((s+".in").c_str(),"r",stdin);
  	freopen((s+".out").c_str(),"w",stdout);
  }
  #define __f(...) 0
  #endif
}

int B, N, D, M;
void solve1(){
	vt<int> p(N);
	rd(p);
	sort(all(p));
	int ans = 0;
	for(auto i : p){
		ans += UB(all(p),i+D) - LB(all(p),i-D) - 1;
	}
	cout << ans/2 << endl;
}

const int MAX_M2 = 75000;
unordered_map<int,int> bit[MAX_M2 + 1];
//const int SHIFT = 1e9;
void solve2(){
	
	auto get_x = [&](int x, int y) -> int {
		return x - y + M;
	};
	
	auto get_y = [&](int x,int y) -> int {
		return x + y - 1;
	};
	
	auto upd = [&](int _x,int _y) -> void {
		//__f("_x,_y",_x,_y);
		for(int x = _x; x < 2 * M; x += (x&(-x))){
			for(int y = _y; y < 2 * M; y += (y&(-y))){
				++bit[x][y];
			}
		}
	};
	
	auto query = [&](int _x,int _y) -> int {
		int ans = 0;
		for(int x = _x; x; x -= (x&(-x))){
			for(int y = _y; y; y -= (y&(-y))){
				ans += bit[x][y];	
			}
		}
		return ans;
	};
	
	vt<pr<int,int>> points;
	f(i,0,N){
		int x, y;
		rd(x,y);
		//x += SHIFT;
		//y += SHIFT;
		points.eb(x,y);
		upd(get_x(x,y), get_y(x,y));
	}
	int ans = 0;
	for(auto [_x,_y] : points){
		int new_x = _x - D, new_y = _y;
		int x = get_x(new_x, new_y), y = get_y(new_x, new_y);
		//__f("x,y",x,y);
		int tmp = query(x+2*D,y+2*D) + query(x-1,y-1) - query(x-1,y+2*D) - query(x+2*D,y-1) - 1;
		//__f("tmp",tmp);
		ans += tmp;
	}
	cout << ans/2 << endl;
}

signed main(){
  setIO();
  rd(B, N, D, M);
  if(B == 1){
  	solve1();
  }else if(B == 2){
  	assert(false);
  	solve2();
  }else{
  	//solve3();
  }
  
}

Subtask #1
6.0 / 6.0

# 결과 실행 시간 메모리 Grader output
1 Correct 5 ms 4300 KB Output is correct
2 Correct 3 ms 4300 KB Output is correct

Subtask #2
6.0 / 6.0

# 결과 실행 시간 메모리 Grader output
1 Correct 4 ms 4428 KB Output is correct

Subtask #3
6.0 / 6.0

# 결과 실행 시간 메모리 Grader output
1 Correct 22 ms 5196 KB Output is correct
2 Correct 21 ms 5196 KB Output is correct

Subtask #4
6.0 / 6.0

# 결과 실행 시간 메모리 Grader output
1 Correct 33 ms 5196 KB Output is correct
2 Correct 26 ms 5196 KB Output is correct

Subtask #5
6.0 / 6.0

# 결과 실행 시간 메모리 Grader output
1 Correct 32 ms 5196 KB Output is correct
2 Correct 29 ms 5196 KB Output is correct
3 Correct 28 ms 5196 KB Output is correct

Subtask #6
0 / 7.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 8 ms 8780 KB Execution killed with signal 6
2 Halted 0 ms 0 KB -

Subtask #7
0 / 7.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 9 ms 8780 KB Execution killed with signal 6
2 Halted 0 ms 0 KB -

Subtask #8
0 / 8.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 8 ms 8796 KB Execution killed with signal 6
2 Halted 0 ms 0 KB -

Subtask #9
0 / 8.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 8 ms 8812 KB Execution killed with signal 6
2 Halted 0 ms 0 KB -

Subtask #10
0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 3 ms 4300 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #11
0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 4 ms 4300 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #12
0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 4 ms 4300 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #13
0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 3 ms 4300 KB Output isn't correct
2 Halted 0 ms 0 KB -