UVA1152-4 Values whose Sum is 0(分塊)
Accept: 794 Submit: 10087
Time Limit: 9000 mSec
Problem Description
The SUM problem can be formulated as follows: given four lists A,B,C,D of integer values, compute how many quadruplet (a,b,c,d) ∈ A×B×C×D are such that a+b+c+d = 0. In the following, we assume that all lists have the same size n.
Input
The input begins with a single positive integer on a line by itself indicating the number of the cases following, each of them as described below. This line is followed by a blank line, and there is also a blank line between two consecutive inputs.
The ?rst line of the input ?le contains the size of the lists n (this value can be as large as 4000). We then have n lines containing four integer values (with absolute value as large as 2^28) that belong respectively to A,B,C and D.
Output
For each test case, your program has to write the number quadruplets whose sum is zero. The outputs of two consecutive cases will be separated by a blank line.Sample Input
16
-45 22 42 -16
-41 -27 56 30
-36 53 -37 77
-36 30 -75 -46
26 -38 -10 62
-32 -54 -6 45
Sample Output
5
題解:這個題主要是太陳了,覺得是個大水題,但是第一次見的時候不是太容易想。思想很深刻,分塊,明明都是暴力枚舉,但即便不加二分查找這個方法也在數量級上碾壓四重for循環,感覺上有一點不可思議,想想莫隊算法是不是也利用了這個思想(分塊真的可以出奇跡)。
1 #include <bits/stdc++.h> 2 3 using namespace std; 4 5 const int maxn = 4000 + 10; 6 7 int a[maxn], b[maxn], c[maxn], d[maxn]; 8 int sum[maxn*maxn]; 9 int n; 10 11 int main() 12 { 13 //freopen("input.txt", "r", stdin); 14 int iCase; 15 scanf("%d", &iCase); 16 while (iCase--) { 17 scanf("%d", &n); 18 for (int i = 0; i < n; i++) { 19 scanf("%d%d%d%d", &a[i], &b[i], &c[i], &d[i]); 20 } 21 22 int cnt = 0; 23 for (int i = 0; i < n; i++) { 24 for (int j = 0; j < n; j++) { 25 sum[cnt++] = a[i] + b[j]; 26 } 27 } 28 sort(sum, sum + cnt); 29 long long ans = 0; 30 for (int i = 0; i < n; i++) { 31 for (int j = 0; j < n; j++) { 32 ans += upper_bound(sum, sum + cnt, -c[i] - d[j]) - lower_bound(sum, sum + cnt, -c[i] - d[j]); 33 } 34 } 35 36 printf("%lld\n", ans); 37 if (iCase) printf("\n"); 38 } 39 return 0; 40 }
UVA1152-4 Values whose Sum is 0(分塊)