CodeForces 817 B.Makes And The Product（水~）

Description

给出$n$个正整数$a_i$，问有多少三元组$(i,j,k)$满足$1\le i<j<k\le n$且$a_i\cdot a_j\cdot a_k$最小

Input

第一行一整数$n$表示序列长度，之后输入$n$个整数$a_i$$(3\le n\le 10^5,1\le a_i\le 10^9)$

Output

输出满足条件的三元组个数

Sample Input

4
1 1 1 1

Sample Output

4

Solution

简单题，最小的一定选，数目多了答案就是组合数

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=100001;
int n;
map<int,int>m;
map<int,int>::iterator it;
ll C(int n,int m)
{
    if(m==1)return n;
    if(m==2)return (ll)n*(n-1)/2;
    if(m==3)return (ll)n*(n-1)*(n-2)/6;
}
int main()
{
    while(~scanf("%d",&n))
    {
        m.clear();
        for(int i=1;i<=n;i++)
        {
            int a;
            scanf("%d",&a);
            m[a]++;
        }
        int i,num=3;
        ll ans;
        for(it=m.begin(),i=0;it!=m.end()&&i<3;it++,i++)
        {
            if(it->second>=num)
            {
                ans=C(it->second,num);
                break;
            }
            else num-=it->second;
        }
        printf("%I64d\n",ans);
    } 
    return 0;
}