【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+;
 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,,sizeof(sum));
     for(int i=;i<=n;i++)
     {
         scanf("%lld",&a[i]);
         m.insert(make_pair(a[i],i));
         sum[i]=sum[i-]+a[i];
     }
     if(sum[n]%==)
     {
         printf("NO\n");
         return ;
     }
     ll half=sum[n]/;
     for(int i=;i<=n;i++)
     {
         ll s=sum[i]-half;
         if(s==)
         {
             printf("YES\n");
             return ;
         }
         else if(s>)
         {
             pos=m.equal_range(s);
             while(pos.first!=pos.second)
             {
                 if(pos.first->second<=i)
                 {
                     printf("YES\n");
                     return ;
                 }
                 pos.first++;
             }
         }
         else
         {
             s=-s;
             pos=m.equal_range(s);
             while(pos.first!=pos.second)
             {
                 if(pos.first->second>i)
                 {
                     printf("YES\n");
                     return ;
                 }
                 pos.first++;
             }
         }
     }
     printf("NO\n");

     return ;
 }

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 << "\n";
        position.first++;
    }
    cout << endl;

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