Cod sursa(job #1998369)

Utilizator MihaelaCismaruMihaela Cismaru MihaelaCismaru Data 7 iulie 2017 16:30:08
Problema Oite Scor 90
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.41 kb
#include<fstream>
#include<vector>
using namespace std;
ifstream in("oite.in");
ofstream out("oite.out");
int n,l,st,dr,mid,k,i,j,oite[1025],csum,alfa,beta,grupuri_de_oite,ok,mod;
vector<pair<int,int> >sume[666014];
vector<int>nr[666014];
int main(){
    in >> n >> l;
    for( i = 1; i <= n; i ++ ){
        in >> oite[i];
    }
    sume[(oite[1] + oite[2] ) %666013].push_back( make_pair (1,2));
    nr[(oite[1] + oite[2] ) %666013 ].push_back(1);
    for( i = 3; i <= n-1; i ++ ){
        for( j = i+1; j <= n; j ++ ){
            csum = l -oite[i] - oite[j];
            mod = csum%666013;
            for( k = 0; k < sume[mod].size(); k ++ ){
                if( oite[sume[mod][k].first] + oite[sume[mod][k].second] == csum ){
                    grupuri_de_oite+= nr[mod][k];
                }
            }

        }
        for( j = 1; j <= i-1; j ++ ){
            ok = 0;
            mod = (oite[i] + oite[j] ) %666013;
            for( k = 0; k < sume[ mod ].size(); k++ ){
                if( oite[sume[mod][k].first] + oite[sume[mod][k].second] == oite[i]+oite[j] ){
                    ok = 1;
                    nr[mod][k]++;
                    break;
                }
            }
            if(ok == 0){
                sume[ mod ].push_back( make_pair(i,j) );
                nr[mod].push_back(1);
            }
        }
    }
    out<<grupuri_de_oite;
    return 0;
}