1. 数值求和
题目大意
给出四个数列,从每个数列中任取一个数问有多少种组合,使这四个数的和为0
思路
若直接枚举则复杂度太高。设 A , B , C , D A,B,C,D A,B,C,D分别来自于四个数列,则 A + B + C + D = 0 < = > A + B = − ( C + D ) A+B+C+D=0 <=> A+B=-(C+D) A+B+C+D=0<=>A+B=−(C+D),所以只需要求A+B的和数列,与C+D的和数列有多少对相反数即可 。在C+D中寻找a的相反数-a的个数时,为了防止遗漏,要先对C+D中的数进行排序,之后,利用二分的办法,分别寻找到-a在数列中第一次出现的位置和最后一次出现的位置,位置做差即可。
代码
#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
using namespace std;
typedef long long llong;
vector<llong>v1;
vector<llong>v2;
int a[4000];
int b[4000];
int c[4000];
int d[4000];
int n;
int s=-1,e=-1;
bool find(int i)
{
s=-1,e=-1;
llong x=-v1[i];
int lv=0,rv=n*n;
while(lv<=rv)
{
int mid=(lv+rv)>>1;
if(x==v2[mid])
{
s=mid;
rv=mid-1;//还要往左找
}
else if(x<v2[mid]){
rv=mid-1;
}
else lv=mid+1;
}
if(s==-1) return false;
//有start必定有end
lv=0;rv=n*n;
while(lv<=rv)
{
int mid=(lv+rv)>>1;
if(v2[mid]==x)
{
e=mid;
lv=mid+1;//还要往右移
}
else if (v2[mid]<x) lv=mid+1;
else if (v2[mid]>x) rv=mid-1;
}
return true;
}
int main()
{
// freopen("in.txt","r",stdin);
scanf("%d",&n);
for(int i=0;i<n;i++)
{
scanf("%lld%lld%lld%lld",a+i,b+i,c+i,d+i);
}
for(int i=0;i<n;i++)
{
for(int j=0;j<n;j++)
{
v1.push_back(a[i]+b[j]);
v2.push_back(c[i]+d[j]);
}
}
int num=0;
sort(v2.begin(),v2.end());
for(int i=0;i<v1.size();i++)
{
if(find(i))
{
num+= e-s+1;
}
}
printf("%d\n",num);
return 0;
}
2. 二分答案求中位数
题目大意
给定一个n个数的数组C,并定义一个新数组:对于任意的i,j,均有
ans[k]=
∣
C
i
−
C
j
∣
| C_i-C_j |
∣Ci−Cj∣,1<=i<j<=n。请求出这个新数组的中位数。C[i]<=1e9,3<=n<=1e5
思路
由于 C i C_i Ci达到1e9,n达到1e5,所以若直接暴力求ans,时间复杂度达到1e10,求中位数也要达到1e9,会超时。 我们需要对答案进行二分,即从0~1e9(C数组的最大差值)中,以二分的方法提出中位数(猜测),再去检测这个提出的中位数的名次;若这个名次小于中位数的名次,则猜测的数需增大;若大于中位数的名次,猜测的数需要减小。增大减小都是二分的过程。
名次计算
- 中位数的名次: C n 2 + 1 2 \frac{C^2_n+1}{2} 2Cn2+1
- 实际名次的计算:需要先对C数组按由大到小进行排序,差值P的名次计算实质上就是计算满足 C j − C i ≤ P , j < i C_j - C_i\leq P, j<i Cj−Ci≤P,j<i的(Ci,Cj)的对数,C数组有序之后,这个式子可以化为 C j ≤ C i + P , j < i C_j \leq C_i+P, j<i Cj≤Ci+P,j<i,我们只需要针对每一个 C i C_i Ci,依次去枚举 C j C_j Cj,就可以获得一个固定的 C i C_i Ci对应的 C j C_j Cj的个数,之后将所有 C i C_i Ci对应的数对个数相加,就可以得到P的名次。枚举 C j C_j Cj的过程,是二分进行的。
代码
注意
有可能枚举出了不存在的差值,但不影响答案:
- 以差值数列: 1 8 9 为例,若枚举的中位数为4,则会得到其before为1,而真正中位数的名次为2,就是说,如果枚举的不是含在里面数(但也可以成为中位数),并且小于中位数的话的话,会永远少一个;
- 若以1 8 10 为例,枚举9,则根据程序,8也一定会被枚举,而且最后答案记住的是8,所以也不必担心枚举出的过大的中位数。
#include<cstdio>
#include<algorithm>
#include<functional>
using namespace std;
typedef long long llong;
int n;
llong c[100099];
llong me_rank=0;
llong check(int p)
{
llong before=0;//p前面有几个数
for(int i=0;i<n;i++)
{
int des=c[i]+p;
int l=0;int r=i-1; //j<i ,cj>ci;
int pos=i;
while(l<=r) //二分,求有多少个满足条件的cj,即cj<=ci+p
{
int m=(l+r)>>1;
if(c[m]<=des)
{
pos=m;
r=m-1; //数越大,越在左边
}
else if(c[m]>des)
{
l=m+1;
}
}
before+=i-1-pos+1;
}
return before; //这就是p的最后出现的位置
}
llong find()
{
llong ans=-1;
llong lv=0,rv=c[0]-c[n-1];
llong rank_mid=0;
llong mid;
//lv,rv最大最小的差值,差值就是索引,对应与中位数的关系
while(lv<=rv)
{
mid=(lv+rv)>>1;
rank_mid=check(mid);
if(me_rank<=rank_mid)
{
ans=mid;
rv=mid-1;
}
else if(me_rank>rank_mid) //当前枚举的数小
{
lv=mid+1;
}
}
return ans;
}
int main()
{
// freopen("in.txt","r",stdin);
while(scanf("%d",&n)!=EOF)
{
for(int i=0;i<n;i++)
{
scanf("%lld",c+i);
}
sort(c,c+n,greater<long long>());
me_rank=(n*(n-1)/2+1)/2; //中位数在整个数列中出现的位置
llong median = find();
printf("%lld\n",median);
}
}