fork download
copy 
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3. #define int              long long int
  4. #define double           long double
  5. inline int power(int a, int b) {
  6.     int x = 1;
  7.     while (b) {
  8.         if (b & 1) x *= a;
  9.         a *= a;
  10.         b >>= 1;
  11.     }
  12.     return x;
  13. }
  14.  
  15.  
  16. const int M = 1000000007;
  17. const int N = 3e5+9;
  18. const int INF = 2e9+1;
  19. const int LINF = 2000000000000000001;
  20.  
  21. //_ ***************************** START Below *******************************
  22.  
  23.  
  24.  
  25. vector<int> a;
  26.  
  27.  
  28. int bruteforce(int n, int k1, int k2){
  29. 	int ans = 0;
  30.  
  31. 	for(int i=0; i<n-2; i++){
  32. 		for(int j=i+1; j<n-2; j++){
  33. 			int leftSum = a[i] + a[j];
  34. 			if(leftSum <= k1) continue;
  35.  
  36. 			for(int k=j+1; k<n-1; k++){
  37. 				for(int l=k+1; l<n; l++){
  38. 					int rightSum = a[k] + a[l];
  39. 					if(rightSum > k2 ) ans++;
  40. 				}
  41. 			}
  42. 		}
  43. 	}
  44. 	return ans;
  45. }
  46.  
  47.  
  48.  
  49.  
  50.  
  51.  
  52. //* Left Count Pre Computation  : 
  53.  
  54. //* Eg : [ 1 , 2 , 3 , 4 , 5 , 6 , 8 ]   k1 = 9
  55.  
  56. //* Ending at analogy : 
  57. //* 	We are finding pairs (x,y) with sum x+y == k1 ending at y
  58.  
  59. //* Observation : if we drew out patterns for each pairs , it looks like this : 
  60. //*      [ (1 , 2 , (3 , (4 , 5) , 6) , 8) ]   
  61. //*     	y = 1,2,3,4 can't form pair with any x such that x+y >= 9
  62. //* 	    y = 5 forms pair with x = 4 only               => ct = 1
  63. //* 		y = 6 forms pair with x = (3, 4, 5)            => ct = 3
  64. //* 		y = 8 forms pair with x = (1, 2, 3, 4, 5, 6)   => ct = 6
  65.  
  66. //* 	Now we can use 2 ptr from 1st pair [x, x+1] such that x + x + 1 >= k1
  67. //* 	                i   j
  68. //*      [ 1 , 2 , 3 , (4 , 5) , 6 , 8) ]   
  69. //* 			p[5] = max(0, 2) = 2
  70. //* 		Now we want to maximize p[5] cts so move i left => i--
  71.  
  72. //* 	           i        j
  73. //*      [ 1 , 2 , 3 , (4 , 5) , 6 , 8) ]   
  74. //* 			5+3 < k1
  75. //* 		Invalid so move to next y => j++
  76.  
  77. //* 	                i       j
  78. //*      [ 1 , 2 , 3 , (4 , 5 , 6) , 8) ]   
  79. //* 			p[6] = max(0, 3) = 3
  80.  
  81. //* 	            i           j
  82. //*      [ 1 , 2 , (3 , 4 , 5 , 6) , 8) ]   
  83. //* 			p[6] = max(3, 4) = 4
  84.  
  85. //* 	        i               j
  86. //*      [ 1 , (2 , 3 , 4 , 5 , 6) , 8) ]   
  87. //* 			Invalid => j++
  88.  
  89. //* O(n + n + n ) = O(n)
  90.  
  91. // int consistency(int n, int k1, int k2) {
  92.  
  93. // 	int ans = 0;
  94.  
  95. // 	vector<int> p(n);
  96.  
  97. // 	vector<int> leftPrefix(n);
  98. // 	int i=1;
  99. // 	while(i<n){
  100. // 		if(a[i] + a[i-1] >= k1) break;
  101. // 		i++;
  102. // 	}
  103. // 	int left = i-1;
  104. // 	int right = i;
  105.  
  106. // 	while(left>=0 && right<n){
  107. // 		if(a[left] + a[right] > k1){
  108. // 			p[right] = max(p[right], right-left);
  109. // 			left--;
  110. // 		}
  111. // 		else {
  112. // 			right++;
  113. // 		}
  114. // 	}
  115. // 	for(int i=1; i<n; i++){
  116. // 		leftPrefix[i] = leftPrefix[i-1] + p[i];
  117. // 	}
  118.  
  119.  
  120.  
  121. // 	vector<int> q(n);
  122.  
  123. // 	vector<int> rightSufix(n);
  124. // 	// for(int i=n-1; i>=0; i--){
  125. // 	// 	rightSufix[i] = rightSufix[i+1] + q[i];
  126. // 	// }
  127.  
  128. // 	for(int j=1; j<n-2; j++){
  129. // 		int leftCt = leftPrefix[j];
  130. // 		int rightCt = rightSufix[j+1]; 
  131.  
  132. // 		ans += leftCt * rightCt;
  133. // 	}
  134.  
  135.  
  136. // 	return ans;
  137.  
  138. // }
  139.  
  140.  
  141.  
  142.  
  143.  
  144.  
  145.  
  146.  
  147.  
  148.  
  149.  
  150.  
  151.  
  152.  
  153.  
  154.  
  155.  
  156. int consistency1(int n, int k1, int k2) {
  157.  
  158. 	//* Suffix 
  159. 	vector<int> pairsInSufix(n);
  160. 	vector<int> pairsFrom(n);
  161.  
  162. 	int s = 0, e = n-1;
  163. 	while(e-s+1>2){
  164. 		if(a[s]+a[e] > k2)	e--;
  165. 		else s++;
  166. 	}
  167. 	int j = e;
  168.  
  169. 	while(s>=0 && e<n){
  170. 		if(a[e] + a[s] > k2){
  171. 			pairsFrom[s] = n-e;
  172. 			s--;
  173. 		}
  174. 		else{
  175. 			e++;
  176. 		}
  177. 	}
  178. 	while(j<n){
  179. 		pairsFrom[j] = n-j-1;
  180. 		j++;
  181. 	}
  182.  
  183. 	pairsInSufix[n-1] = pairsFrom[n-1];
  184. 	for(int i=n-2; i>=0; i--){
  185. 		pairsInSufix[i] = pairsFrom[i] + pairsInSufix[i+1];
  186. 	}
  187.  
  188.  
  189. 	int ans = 0;
  190.  
  191. 	//* Global 2 Ptr * local 2 Ptr 
  192. 	for(int j=n-3, i=0; j>i; j--){
  193. 		while(i<j && a[i]+a[j] <= k1) i++;
  194.  
  195. 		int left = j-i;
  196. 		int right = pairsInSufix[j+1];
  197.  
  198. 		ans += left*right;
  199. 	}
  200.  
  201. 	return ans;
  202. }
  203.  
  204.  
  205.  
  206.  
  207.  
  208.  
  209. int consistency2(int n, int k1, int k2) {
  210.  
  211. 	//* prefix 
  212. 	vector<int> pairsInPrefix(n);
  213. 	vector<int> pairsEndingAt(n);
  214.  
  215. 	int s = 0, e = n-1;
  216. 	while(e-s+1>2){
  217. 		if(a[s]+a[e] > k1)	e--;
  218. 		else s++;
  219. 	}
  220.  
  221. 	while(s>=0 && e<n){
  222. 		if(a[e] + a[s] > k1){
  223. 			s--;
  224. 		}
  225. 		else{
  226. 			pairsEndingAt[e] = e-s-1;
  227. 			e++;
  228. 		}
  229. 	}
  230. 	while(e<n){
  231. 		pairsEndingAt[e] = e;
  232. 		e++;
  233. 	}
  234.  
  235.  
  236.  
  237.  
  238.  
  239. 	//* Suffix 
  240.  
  241. 	vector<int> pairsInSufix(n);
  242. 	vector<int> pairsFrom(n);
  243.  
  244. 	s = 0, e = n-1;
  245.  
  246. 	while(e-s+1>2){
  247. 		if(a[s]+a[e] > k2)	e--;
  248. 		else s++;
  249. 	}
  250. 	int y = e;
  251.  
  252. 	while(s>=0 && e<n){
  253. 		if(a[e] + a[s] > k2){
  254. 			pairsFrom[s] = n-e;
  255. 			s--;
  256. 		}
  257. 		else{
  258. 			e++;
  259. 		}
  260. 	}
  261. 	while(y<n){
  262. 		pairsFrom[y] = n-y-1;
  263. 		y++;
  264. 	}
  265.  
  266. 	pairsInSufix[n-1] = pairsFrom[n-1];
  267. 	for(int i=n-2; i>=0; i--){
  268. 		pairsInSufix[i] = pairsFrom[i] + pairsInSufix[i+1];
  269. 	}
  270.  
  271.  
  272. 	int ans = 0;
  273. 	for(int i=1; i<n-2; i++){
  274. 		int left = pairsEndingAt[i];
  275. 		int right = pairsInSufix[i+1];
  276.  
  277. 		ans += left * right;
  278. 	}
  279.  
  280. 	return ans;
  281. }
  282.  
  283.  
  284.  
  285.  
  286.  
  287.  
  288.  
  289.  
  290.  
  291.  
  292.  
  293.  
  294. int practice(int n, int k1, int k2) {
  295.  
  296. 	return 0;
  297.  
  298. }
  299.  
  300.  
  301.  
  302.  
  303.  
  304.  
  305.  
  306.  
  307.  
  308.  
  309. void solve() {
  310.  
  311. 	int n, k1, k2;
  312. 	cin >> n >> k1 >> k2;
  313.  
  314. 	a.resize(n);
  315. 	for(int i=0; i<n; i++) cin >> a[i];
  316.  
  317.     cout << bruteforce(n, k1, k2) << " ";
  318.     cout << consistency1(n, k1, k2) << " ";
  319.     cout << consistency2(n, k1, k2) << endl;
  320.  
  321.  
  322.     // cout << bruteforce(n, k1, k2) << " -> ";
  323.     // cout << practice(n, k1, k2) << endl;
  324.  
  325. }
  326.  
  327.  
  328.  
  329.  
  330.  
  331. int32_t main() {
  332.     ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
  333.  
  334.     int t = 1;
  335.     cin >> t;
  336.     while (t--) {
  337.         solve();
  338.     }
  339.  
  340.     return 0;
  341. }
Success #stdin #stdout 0s 5320KB
comments (?)
stdin
copy 
4
6 2 2
1 1 1 1 2 2
6 1 3
1 1 1 1 2 2
8 9 7
1 2 3 4 5 6 8 12
6 1 3
0 1 1 1 2 2
6 3 3
1 2 2 2 2 3
stdout
copy 
0 0 0
6 6 6
2 2 2
3 3 3
https://ideone.com/HUR6FO
language:
C++14 (gcc 8.3)
created:
3 days ago
visibility:
public

Share or Embed source code

Discover > Sphere Engine API

The brand new service which powers Ideone!

Discover > IDE Widget

Widget for compiling and running the source code in a web browser!