【multimap的应用】D. Array Division

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

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cmath>
#include<set>
#include<map> 
#include<utility>
using namespace std;
typedef long long ll;
const int maxn=1e5+2;
ll a[maxn];
ll sum[maxn];
multimap<ll,int> m;
multimap<ll,int> ::iterator it;
pair<multimap<ll,int>::iterator,multimap<ll,int>::iterator> pos; 
int main()
{
    int n;
    scanf("%d",&n);
    memset(sum,0,sizeof(sum));
    for(int i=1;i<=n;i++)
    {
        scanf("%lld",&a[i]);
        m.insert(make_pair(a[i],i)); 
        sum[i]=sum[i-1]+a[i];
    }
    if(sum[n]%2==1)
    {
        printf("NO
");
        return 0;
    }
    ll half=sum[n]/2;
    for(int i=1;i<=n;i++)
    {
        ll s=sum[i]-half;
        if(s==0)
        {
            printf("YES
");
            return 0;
        }
        else if(s>0)
        {
            pos=m.equal_range(s);
            while(pos.first!=pos.second)    
            {
                if(pos.first->second<=i)
                {
                    printf("YES
");
                    return 0;
                }
                pos.first++;
            }
        }
        else 
        {
            s=-s;
            pos=m.equal_range(s);
            while(pos.first!=pos.second)
            {
                if(pos.first->second>i)
                {
                    printf("YES
");
                    return 0;
                }
                pos.first++;
            }
        }
    }
    printf("NO
");
    
    return 0;
}

equal_range的用法：

 //定义pair 对象position; pair数据类型是 2个 multimap<string,string>::iterator 指针。
    pair<multimap<string, string>::iterator, multimap<string, string>::iterator> position;
    //如果键存在，函数返回2个指针，第一个指针指向键第一个匹配的元素
    //第二个指针指向键最后一个匹配的元素的下一位置
    position = author.equal_range(search_item);
    while (position.first != position.second)
    {
        cout << position.first->first << "  " << position.first->second << " ";
        position.first++;
    }
    cout << endl;

map的find函数只能找key,所以需要把a[i]作为key,但同一个数可以出现多次，所以需要用multimap,multimap内容根据key排序后，equal_range函数key出现的区间。