UVA10125 Sumsets

嘟嘟嘟




很简单的折半搜索。

把式子变一下型，得到\(a + b = d - c\)。

然后枚举\(a, b\)，存到\(map\)里，再枚举\(c, d\)就好了。

\(map\)以\(a,b\)两数之和为下标。为了判重，\(map\)的第二个参数是一个\(vector\)，\(vector\)里面又存了两个数\(a, b\)。

这样先\(O(n ^ 2)\)跑一边\(a, b\)之和，存到\(map\)里，然后从大到小枚举\(d\)和\(c\)，遍历\(map\)中\(d - c\)的\(vector\)，如果四个数都没有一样的，就直接返回好了。

用\(map\)的最大好处是代码特别短。

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdlib>
#include<cctype>
#include<vector>
#include<stack>
#include<queue>
#include<map>
using namespace std;
#define enter puts("")
#define space putchar(' ')
#define Mem(a, x) memset(a, x, sizeof(a))
#define rg register
typedef long long ll;
typedef double db;
const int INF = 0x3f3f3f3f;
const db eps = 1e-8;
const int maxn = 1e3 + 5;
inline ll read()
{
  ll ans = 0;
  char ch = getchar(), last = ' ';
  while(!isdigit(ch)) last = ch, ch = getchar();
  while(isdigit(ch)) ans = (ans << 1) + (ans << 3) + ch - '0', ch = getchar();
  if(last == '-') ans = -ans;
  return ans;
}
inline void write(ll x)
{
  if(x < 0) x = -x, putchar('-');
  if(x >= 10) write(x / 10);
  putchar(x % 10 + '0');
}

int n, a[maxn];

struct Node
{
  int x, y;
};
map<int, vector<Node> > mp;

int solve()
{
  for(int i = n; i; --i)
      for(int j = 1; j < i; ++j)
	{
	  int sum = a[i] - a[j];
	  for(int k = 0; k < (int)mp[sum].size(); ++k)
	    {
	      int x = mp[sum][k].x, y = mp[sum][k].y;
	      if(a[i] != x && a[i] != y && a[j] != x && a[j] != y)
		return a[i];
	    }
	}
  return 536870912;
}

int main()
{
  while(scanf("%d", &n) && n)
    {
      mp.clear(); Mem(a, 0);
      for(int i = 1; i <= n; ++i) a[i] = read();
      sort(a + 1, a + n + 1);
      for(int i = 1; i < n; ++i)
	for(int j = i + 1; j <= n; ++j)
	  mp[a[i] + a[j]].push_back((Node){a[i], a[j]});
      int ans = solve();
      if(ans == 536870912) puts("no solution");
      else write(ans), enter;
    }
  return 0;
}