【Codeforces】808D Array Division(前后两部分和相等)

http://codeforces.com/contest/808/problem/D

给你一个数组,问:是否可以通过移动一个数字的位置,求只能移动一次,使得这个数组前后部分的和相等,前后部分不一定等长

一个a数组储存数据,另一个b数组b[i]表示前i项和

如过存在这样的数,可以让前后的和相等,那么在没有移动之前,前后部分的差的绝对值等于这个数的两倍,所以前后部分差的绝对值一定是一个偶数

加上map映射

#include <iostream>
#include <cstring>
#include <map>
using namespace std;

typedef long long ll;

ll a[100000+5];
ll b[100000+5];
map<ll,ll> m1,m2;

int main()
{
	ll n;
	cin >> n;
	ll i;
	b[0] = 0;
	for(i=1;i<=n;i++)
	{
		cin >> a[i];
		m2[a[i]]++;//把所有的数都先放到m2中 
		b[i] = b[i-1] + a[i]; 
	}
	if(b[n] & 1)
	{
		cout << "NO";
		return 0;
	}
	for(i=1;i<=n;i++)
	{
		m1[a[i]]++;
		m2[a[i]]--;//从m2中取出,放入m1 
		ll s1 = b[i];
		ll s2 = b[n]-b[i];
		if(s1==s2)
		{
			cout << "YES";
			return 0;
		}
		if(s2-s1>0 && (s2-s1)%2==0 && m2[(s2-s1)/2]>0)//相差得那个数一定是在和较大的那一个部分中 
		{
			cout << "YES";
			return 0;
		}
		if(s1-s2>0 && (s1-s2)%2==0 && m1[(s1-s2)/2]>0)
		{
			cout << "YES";
			return 0;
		}
	}
	cout << "NO";
	return 0;
}

 

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值