Cod sursa(job #1005350)

Utilizator andrettiAndretti Naiden andretti Data 4 octombrie 2013 20:53:01
Problema Oite Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.08 kb
#include<stdio.h>
#include<vector>
#define pb push_back
#define mp make_pair
#define maxn 1030
#define MOD 666013
using namespace std;

int n,S,sol;
int a[maxn];
vector< pair<int,int> > H[MOD];

void read()
{
    scanf("%d%d",&n,&S);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
}

int hash_key(int k){
    return k%MOD;
}

void insert(int val)
{
    int k=hash_key(val);
    for(unsigned int i=0;i<H[k].size();i++)
     if(H[k][i].first==val) {H[k][i].second++;return;}
    H[k].pb(mp(val,1));
}

int search(int val)
{
    if(val<0) return 0;
    int k=hash_key(val);
    for(unsigned int i=0;i<H[k].size();i++)
     if(H[k][i].first==val) return H[k][i].second;
    return 0;
}

void solve()
{
    for(int i=1;i<n;i++)
    {
        for(int j=i+1;j<=n;j++)
         sol+=search(S-a[i]-a[j]);
        for(int j=1;j<i;j++) insert(a[i]+a[j]);
    }
}

int main()
{
    freopen("oite.in","r",stdin);
    freopen("oite.out","w",stdout);

    read();
    solve();
    printf("%d",sol);

    fclose(stdin);
    fclose(stdout);
    return 0;
}