Cod sursa(job #332236)

Utilizator DranaXumAlexandru Dumitru Paunoiu DranaXum Data 17 iulie 2009 01:14:18
Problema Numarare triunghiuri Scor 45
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.91 kb
#include<fstream>
#include<vector>
#include<algorithm>

using namespace std;

ifstream fin("nrtri.in");
ofstream fout("nrtri.out");

int poz[30001];
int ap[30001];
int cnt[30001];
vector<int> nrs;

int main()
{
    int sol=0,n,i,j,x;
    fin>>n;
    for(i=0;i<n;i++) {fin>>x; nrs.push_back(x);}

    make_heap(nrs.begin(),nrs.end());
    sort_heap(nrs.begin(),nrs.end());


    for(i=0;i<n;i++) {if(!poz[nrs[i]]) poz[nrs[i]]=i+1; ap[nrs[i]]++;}
    for(i=1;i<=30000;i++)
    {    if(poz[i])
            cnt[i]=cnt[i-1]+ap[i];
        else cnt[i]=cnt[i-1];
    }
    for(i=0;i<n;i++){
       for(j=i+1;j<n;j++)
        {
            int pozi=poz[nrs[i]+nrs[j]];
            sol+=cnt[nrs[i]+nrs[j]]-cnt[nrs[j]];
            if(j-poz[nrs[j]]+2<=ap[nrs[j]])
                sol+=ap[nrs[j]]-j+poz[nrs[j]]-2;
        }
    }
    fout<<sol;
    fin.close();
    fout.close();
    return 0;
}