Submission #1769646

---

Source Code Expand

#include <iostream>
using namespace std;



unsigned int *initial_sort(unsigned int &max);
void cal (unsigned int *a1, unsigned int *a2 ,unsigned int *b, unsigned int max);
void cal (unsigned int *a1, unsigned int *a2, unsigned int max);
unsigned int sum (unsigned int *a, unsigned int *b,unsigned int max);

int main(void) {
    unsigned int max = 1;
    unsigned int *a1;
    a1 = initial_sort(max);
    unsigned int *a2 = new unsigned int[(max/2)/2+1];
    unsigned int *a3 = new unsigned int[max/2+1];
    unsigned int *a4 = new unsigned int[max+1];

    cal (a1,a2,(max/2)/2);
    cal (a2,a3,a1,max/2);
    cal (a3,a4,a1,max);

    cout << sum (a4,a1,max) << endl;

    return 0;
}

void cal (unsigned int *a1, unsigned int *a2 ,unsigned int *b, unsigned int max)
{
    a2[0] = a2[1] = 0;
    for (unsigned int i = 2;i <= max;i ++){
        a2[i] = (a2[i-2] + (a1[i/2]*b[i/2])%1000000007)%1000000007;
    }
}

void cal (unsigned int *a1, unsigned int *a2, unsigned int max)
{
    a2[0] = a2[1] = 0;
    for (unsigned int i = 2;i <= max;i++){
        a2[i] = (a2[i-2] + a1[i/2])%1000000007;
    }
}

unsigned int sum (unsigned int *a, unsigned int *b, unsigned int max)
{
    unsigned int result = 0;
    for (unsigned int i = 0;i <= max;i ++){
        result = (result + (a[i]*b[i])%1000000007)%1000000007;
    }
    return result;
}

unsigned int *initial_sort(unsigned int &max)
{
    unsigned int N,M;
    cin >> N;
    unsigned int *init = new unsigned int[1+(unsigned int)10e5];
    for (unsigned int i = 0;i <= 10e5;i ++) init[i] = 0;
    for (unsigned int i = 0;i < N;i ++){
        cin >> M;
        if (max < M) max = M;
        init[M] ++;
    }
    return init;
}

Submission Info

Judge Result