mickypinata

PROG-T1090: Cock Tail

Jun 12th, 2021
211
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 0.70 KB | None | 0 0
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3.  
  4. typedef long long lli;
  5.  
  6. const int T = 1e5;
  7.  
  8. int chicken[T + 10];
  9.  
  10. int main(){
  11.  
  12.     int nChicken;
  13.     int mx = 0;
  14.     scanf("%d", &nChicken);
  15.     for(int i = 1; i <= nChicken; ++i){
  16.         int x;
  17.         scanf("%d", &x);
  18.         ++chicken[x];
  19.         mx = max(mx, x);
  20.     }
  21.  
  22.     int tr;
  23.     scanf("%d", &tr);
  24.     lli cnt = 0;
  25.     for(int i = 0; i <= tr >> 1; ++i){
  26.         if(tr - i > mx){
  27.             continue;
  28.         }
  29.         if(i + i == tr){
  30.             cnt += (lli)chicken[i] * (chicken[i] - 1) / 2;
  31.         } else {
  32.             cnt += (lli)chicken[i] * chicken[tr - i];
  33.         }
  34.     }
  35.     cout << cnt;
  36.  
  37.     return 0;
  38. }
  39.  
Add Comment
Please, Sign In to add comment