天天看点

hdu6168 Numbers 2017多校1008 map

http://acm.split.hdu.edu.cn/showproblem.php?pid=6168

题意:由一个a数组,通过ai+aj(1<=i<j<=n)构成b数组。现在给你a和b数组的混合,求a数组。

题解:每次存入时,用map标记每个数字的出现次数(数据是从小到大给的,所以可以不用排序)。然后前两个数开始相加,然后更新map(感觉有点筛法的意思)。更新之后,如果map中这个数还出现,就记入答案并更新map。

代码:

#include<bits/stdc++.h>
#define debug cout<<"aaa"<<endl
#define d(a) cout<<a<<endl
#define mem(a,b) memset(a,b,sizeof(a))
#define LL long long
#define lson l,mid,root<<1
#define rson mid+1,r,root<<1|1
#define MIN_INT (-2147483647-1)
#define MAX_INT 2147483647
#define MAX_LL 9223372036854775807i64
#define MIN_LL (-9223372036854775807i64-1)
using namespace std;

const int N = 500 + 5;
const int mod = 1000000000 + 7;
const double eps = 1e-8;
int a[130000];
map<int,int> m;

int main(){
	int n,ans[N],cnt;
	while(~scanf("%d",&n)){
		m.clear(),mem(ans,0),cnt=3;
		for(int i=1;i<=n;i++){
			scanf("%d",&a[i]);
			m[a[i]]++;
		}
		ans[1]=a[1],ans[2]=a[2];
		m[a[1]]--,m[a[2]]--,m[a[1]+a[2]]--;
		for(int i=3;i<=n;i++){
			if(m[a[i]]!=0){//计入答案,并更新map
				ans[cnt]=a[i];
				m[a[i]]--;
				for(int j=1;j<cnt;j++){
					m[ans[cnt]+ans[j]]--;
				}
				cnt++;
			}
		}
		printf("%d\n",cnt-1);
		for(int i=1;i<cnt-1;i++){
			printf("%d ",ans[i]);
		}
		printf("%d\n",ans[cnt-1]);
	}
	return 0;
}