UVA11524 Values whose Sum is 0 题解

q779

已于 2022-07-26 09:32:40 修改

阅读量117

点赞数 1

分类专栏： OI 文章标签：算法数据结构

于 2021-08-31 23:02:47 首次发布

本文链接：https://blog.csdn.net/qq_50332374/article/details/120028934

版权

OI 专栏收录该内容

240 篇文章 3 订阅

订阅专栏

UVA1152 4 Values whose Sum is 0

题目链接：UVA1152 4 Values whose Sum is 0

题意：给定四个数组，求 $a_i+b_j+c_k+d_l = 0$ 的个数，多组数据+奇怪的输出格式

解法一

$O(n^4)$ 枚举肯定过不去

想一想，我们有必要在枚举 $i, j$ 的时候再去枚举 $k, l$ 吗？

没有，因为 $k, l$ 的组合是有限的，只有这么多

我们可以先 $O(n^2)$ 把所有 $c_k+d_l$ 枚举出来，然后排个序，就可以二分找出所有的 $a_i+b_j)$ 了

时间复杂度 $O(n^2\log n)$

代码如下

#include <bits/stdc++.h>
using namespace std;
#define int long long
#define R register
#define MAXN (int)(4e3+5)
template<typename T>inline void read(R T &k)
{
	R char ch=getchar();R T x=0,f=1;
	while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
	while(isdigit(ch)){x=(x<<1)+(x<<3)+(ch^48);ch=getchar();}
	k=x*f;
}
template<typename T>inline void write(R T k)
{
	if(k<0)putchar('-'),k=-k;
	if(!k){putchar('0');return;}
	if(k>9)write(k/10);
	putchar(k%10+'0');
}
int Q,n,ans;
int a[MAXN],b[MAXN],c[MAXN],d[MAXN],t[MAXN*MAXN+MAXN];
void solve()
{
	ans=0;read(n);
	for(R int i=1; i<=n; i++)
	{
		read(a[i]);read(b[i]);
		read(c[i]);read(d[i]);
	}
	for(R int i=1; i<=n; i++)
		for(R int j=1; j<=n; j++)
			t[(i-1)*n+j]=c[i]+d[j];
	sort(t+1,t+n*n+1);
	for(R int i=1; i<=n; i++)
		for(R int j=1; j<=n; j++)
		{
			R int tmp=a[i]+b[j];
			ans+=upper_bound(t+1,t+n*n+1,-tmp)-lower_bound(t+1,t+n*n+1,-tmp);
		}
	write(ans);puts("");
	if(Q)puts("");
}
signed main()
{
	read(Q);
	while(Q--)solve();
	return 0;
}