2016级算法第二次上机-D.Bamboo的饼干

Bamboo的饼干

分析

从两个数组中各取一个数，使两者相加等于给定值。要注意去重和排序

难度不大，方法很多，基本只要不大于O(n^2 ) 的都可以过。本意想考察二分搜索

还可以借助stl中的map，set以及lower_bound等，当然只用数组也可以做。由于数据范围不大，也可以直接用数组下标来计数。

提起去重，有同学似乎一直纠结 （2,3）和（3,2）算不算重复数对..不算!只有（2,1）（2,1）这样的是真·重复对

map

这是很多AC代码用到的方法。因为map的key值是不重复且有序的，因此很适合本题。

参考代码

int n, t, x;
map<int, int> m;
int A[MaxSize];

int main()
{
    while(~scanf("%d", &n))
    {
        m.clear();
        int ans = 0;
        for (int i = 0; i < n; ++i) {
            scanf("%d", &x);
            m[x] = 1;
        }
        for (int i = 0; i < n; ++i) {
            scanf("%d", &A[i]);
        }
        scanf("%d", &t);
        sort(A, A+n);

        for (int i = n-1; i >= 0; --i) {
            long long tem = (long long)t - (long long)A[i];
            if(m[tem] > 0)//查询方便
            {
                ans++;
                printf("%lld %d\n", tem, A[i]);
                m[tem] = 0;
            }
        }

        if(ans == 0) printf("OTZ\n");
        printf("\n");
    }
}

数组

此处@周宏建，数组下标计数的方式，与上面map功能相似，注意map键值可为负但是数组下标不可以。当数据范围过大时该方法可能受限。

下面是这位同学上机时的AC代码：

#define bias 10000001
using namespace std;
bool a[20000010];
int b[100005];
int main()
{
	int n,x,target;
	while(~scanf("%d",&n))
	{
		memset(a,0,sizeof(a));
		bool flag=false;
		for(int i=0;i<n;++i)
		{
			scanf("%d",&x);
			a[x+bias]=1;
		}
		for(int i=0;i<n;++i)
		scanf("%d",b+i);
		scanf("%d",&target);
		sort(b,b+n);
		int tmp;
		for(int i=n-1;i>=0;--i)
		{
			tmp=target-b[i]+bias;
			if(tmp>=0&&tmp<20000010&&a[tmp])
			{
				a[tmp]=0;
				flag=true;
				printf("%d %d\n",target-b[i],b[i]);
			}
		}
		if(!flag)
		printf("OTZ\n");
		printf("\n");
	}
}

数组+二分

这也是非常常见的思路。因为只有两个数，确定一个查找另一个，本质就是个查找。当普通查找TLE时应当会想到用二分查找来做。手写和STL均可。

参考代码

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<vector>
using namespace std;
const int maxx = 100004;
int a[maxx],b[maxx];
int BinarySearch(int a[], int l ,int r,int val)
{
    int mid  ;
    while(l<=r)
    {
        mid = (l+r)/2;
        if(a[mid]==val)return mid;
        else if(a[mid]>val)r = mid-1;
        else l  = mid+1;
    }
    return -1;
}
int main()
{
   int n,t,tt,pos;
   while(~scanf("%d",&n))
   {
       for(int i = 0;i<n;i++)
        scanf("%d",&a[i]);
       for(int j = 0;j<n;j++)
        scanf("%d",&b[j]);
       scanf("%d",&t);
       sort(a,a+n);sort(b,b+n);
       bool flag = false;
       for(int i = 0;i<n;i++)
       {
           if(i>0&&a[i]==a[i-1])continue;
           tt = t-a[i];
           pos = BinarySearch(b,0,n-1,tt);
           if(pos>-1&&pos<n)
           {
               printf("%d %d\n",a[i],b[pos]);
               flag = true;
           }
       }
       if(flag==false)printf("OTZ\n");
       printf("\n");
   }
}

当然，因为查找一个数，哈希表还能更快

拓展

请大家思考一下，如果给一个数组找三个数之和为某一值呢？四个数之和呢？