lina_os

Untitled

Mar 13th, 2025
8
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.92 KB | None | 0 0
  1. #include <bits/stdc++.h>
  2. #define ll long long
  3.  
  4. using namespace std;
  5.  
  6. int main() {
  7. ios_base::sync_with_stdio(false), cin.tie(nullptr), cout.tie(nullptr);
  8. int t;
  9. cin >> t;
  10. while (t--) {
  11. int n;
  12. cin >> n;
  13. vector<ll>v(n);
  14. ll a[n];
  15. for (int i=0; i<n; i++) {
  16. cin >> a[i];
  17. v[i]=a[i]*a[i]-a[i]*(i+1)+(i+1)*(i+1);
  18. }
  19. sort(v.begin(), v.end());
  20. ll ans=0;
  21. ll x=1;
  22. for (int i=1; i<n; i++) {
  23. if (v[i]!=v[i-1]) {
  24. ans+=x*(x-1)/2;
  25. x=1;
  26. }
  27. else x++;
  28. }
  29. ans+=x*(x-1)/2;
  30. cout << ans << endl;
  31. }
  32. return 0;
  33. }
  34.  
  35.  
  36. /* aj^2-aj.j+j^2 == ai^2-.ai.i+i^2
  37. * aj=ai && j=i XXX
  38. * aj=i && ai=j ---
  39. * aj2-aj.j=ai2-ai.i && i=j XXX
  40. * aj=ai && j(j-aj)=i(i-ai)
  41. * aj=i && j(j-aj)=ai(ai-i)
  42. * ai=j && i(i-ai)=aj(aj-j)
  43. */
Advertisement
Add Comment
Please, Sign In to add comment