prietenie

Time limit: 0.1s Memory limit: 64MB Input: prietenie.in Output: prietenie.out

Elevii celor două clase de a șaptea din școală merg în excursie. În fiecare clasă sunt câte NN elevi. Ovidiu și Mihnea, fiind liderii celor două clase din care fac parte, doresc să analizeze reușita excursiei, în funcție de gradul de compatibilitate dintre elevii participanți la excursie.

Pentru a determina acest grad, fiecărui elev din cele două clase îi este atribuit un coeficient de amabilitate. Astfel, elevii din clasa lui Ovidiu au, în ordinea din catalog, coeficienții a1,a2,,aNa_1, a_2, \dots, a_N, iar elevii din clasa lui Mihnea au, în ordinea din catalog, coeficienții b1,b2,,bNb_1, b_2, \dots, b_N.

Gradul de compatibilitate dintre doi elevi din clase diferite este definit ca pătratul diferenței dintre coeficienții de amabilitate atribuiți fiecăruia. Astfel, gradul de compatibilitate GijG_{ij} dintre al ii-lea elev din clasa lui Ovidiu și al jj-lea elev din clasa lui Mihnea este egal cu (aibj)2(a_i - b_j)^2, cu 1iN1 \leq i \leq N și 1jN1 \leq j \leq N.

Gradul de compatibilitate dintre cele două clase este suma tuturor gradelor de compatibilitate dintre oricare doi elevi din clase diferite, adică suma tuturor valorilor GijG_{ij} cu 1iN1 \leq i \leq N și 1jN1 \leq j \leq N.

Pentru a lega o prietenie durabilă doi elevi din clase diferite trebuie să aibă gradul de compatibilitate fie mai mic sau egal cu XX, fie mai mare sau egal cu YY, unde XX și YY sunt valori date (adică GijXG_{ij} \leq X sau YGijY \leq G_{ij})

Se cunosc NN, coeficienții a1,a2,,aNa_1, a_2, \dots, a_N și b1,b2,,bNb_1, b_2, \dots, b_N, precum și valorile XX și YY, cu semnificația din enunț.

Cerință

  • Determinați gradul de compatibilitate dintre cele două clase.
  • Determinați, pentru fiecare elev din clasa lui Ovidiu, numărul de elevi din clasa lui Mihnea cu care acesta poate lega o prietenie durabilă.

Date de intrare

Fișierul prietenie.in conține pe prima linie un singur număr natural CC, semnificând cerința care trebuie rezolvată (care poate fi doar 1 sau 2).
Pe a doua linie se găsesc trei numere naturale NN, XX și YY, cu semnificația din enunț.
Pe a treia linie se găsesc NN numere naturale a1,a2,,aNa_1, a_2, \dots, a_N, cu semnificația din enunț.
Pe a patra linie se găsesc NN numere naturale b1,b2,,bNb_1, b_2, \dots, b_N, cu semnificația din enunț.
Numerele aflate pe aceeași linie a fișierului sunt separate prin câte un spațiu.

Date de ieșire

Fișierul prietenie.out conține:

  • dacă C=1C = 1, numărul natural determinat pentru cerința 11;
  • dacă C=2C = 2, NN numere naturale, separate prin câte un spațiu, reprezentând numerele determinate pentru cerința 22, corespunzătoare ordinii în care elevii apar în catalogul clasei.

Restricții și precizări

  • 1N200 0001 \leq N \leq 200 \ 000.
  • 0ai,bj30 0000 \leq a_i, b_j \leq 30 \ 000, pentru oricare 1i,jN1 \leq i, j \leq N.
  • 0XY900 000 0000 \leq X \leq Y \leq 900 \ 000 \ 000.
  • Ovidiu a observat că (aibj)2(a_i - b_j)^2 se poate scrie și sub forma ai2+bj22aibja_i^2 + b_j^2 - 2 \cdot a_i \cdot b_j.
# Scor Restricții
1 25 C=1C = 1 și 1N2 5001 \leq N \leq 2 \ 500
2 10 C=1C = 1, 2 500<N200 0002 \ 500 < N \leq 200 \ 000 și 0ai,bj5 0000 \leq a_i, b_j \leq 5 \ 000, pentru oricare 1i,jN1 \leq i, j \leq N
3 10 C=1C = 1 și 2 500<N200 0002 \ 500 < N \leq 200 \ 000
4 25 C=2C = 2 și 1N2 5001 \leq N \leq 2 \ 500
5 15 C=2C = 2, 2 500<N200 0002 \ 500 < N \leq 200 \ 000 și 0ai,bj5 0000 \leq a_i, b_j \leq 5 \ 000, pentru oricare 1i,jN1 \leq i, j \leq N
6 15 C=2C = 2 și 2 500<N200 0002 \ 500 < N \leq 200 \ 000

Exemplul 1

prietenie.in

1
4 3 10
1 3 5 7
5 1 4 2

prietenie.out

136

Explicație

Se rezolvă cerința C=1C = 1.

Avem N=4N = 4.

  • G11=(a1b1)2=(15)2=16G_{11} = (a_1 - b_1)^2 = (1 - 5)^2 = 16;
  • G12=(a1b2)2=(11)2=0G_{12} = (a_1 - b_2)^2 = (1 - 1)^2 = 0;
  • \dots

Gradul de compatibilitate dintre cele două clase este egal cu:
(15)2+(11)2+(14)2+(12)2+(1 - 5)^2 + (1 - 1)^2 + (1 - 4)^2 + (1 - 2)^2 +
(35)2+(31)2+(34)2+(32)2+(3 - 5)^2 + (3 - 1)^2 + (3 - 4)^2 + (3 - 2)^2 +
(55)2+(51)2+(54)2+(52)2+(5 - 5)^2 + (5 - 1)^2 + (5 - 4)^2 + (5 - 2)^2 +
(75)2+(71)2+(74)2+(72)2=(7 - 5)^2 + (7 - 1)^2 + (7 - 4)^2 + (7 - 2)^2 =
16+0+9+1+4+4+1+1+0+16+1+9+4+36+9+25=13616 + 0 + 9 + 1 + 4 + 4 + 1 + 1 + 0 + 16 + 1 + 9 + 4 + 36 + 9 + 25 = 136

Exemplul 2

prietenie.in

2
4 3 10
1 3 5 7
5 1 4 2

prietenie.out

3 2 3 2

Explicație

Se rezolvă cerința C=2C = 2, pentru care N=4N = 4, X=3X = 3 și Y=10Y = 10.

Pentru primul elev din clasa lui Ovidiu, care are coeficientul a1=1a_1 = 1, gradele de compatibilitate cu elevii din cealaltă clasă sunt:

  • (a1b1)2=(15)2=16(a_1 - b_1)^2 = (1 - 5)^2 = 16, iar 16Y16 \geq Y;
  • (a1b2)2=(11)2=0(a_1 - b_2)^2 = (1 - 1)^2 = 0, iar 0X0 \leq X;
  • (a1b3)2=(14)2=9(a_1 - b_3)^2 = (1 - 4)^2 = 9, iar X<9<YX < 9 < Y;
  • (a1b4)2=(12)2=1(a_1 - b_4)^2 = (1 - 2)^2 = 1, iar 1X1 \leq X;

Astfel, el poate lega o prietenie de lungă durata cu 33 elevi din clasa lui Mihnea: cu primul, cu al doilea și cu al patrulea.

Al doilea elev din clasa lui Ovidiu poate lega o prietenie de lungă durata cu doi elevi din clasa lui Mihnea: cu al treilea și cu al patrulea.
Analog, al treilea și al patrulea elev din clasa lui Ovidiu pot lega o prietenie de lungă durată cu 3 elevi, respectiv cu 2 elevi din clasa lui Mihnea.

Log in or sign up to be able to send submissions!