fork download
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3.  
  4. #define int long long
  5. void solve(){
  6. int n,c;
  7. cin>>n>>c;
  8. vector<int> v(n);
  9. for(int i=0;i<n;i++){
  10. cin>>v[i];
  11. }
  12. int a=0;
  13. for(int i=0;i<n;i++){
  14. a+=(v[i]*v[i]);
  15. }
  16. int b=0;
  17. for(int i=0;i<n;i++){
  18. b+=v[i];
  19. }
  20. int k=b;
  21. a=n*(a-c);
  22. b=(b*b);
  23. int k2=sqrt(b*b-a);
  24. cout<<(k2-b)%n+((k2-b)/n)<<endl;
  25.  
  26.  
  27.  
  28.  
  29.  
  30. }
  31. signed main() {
  32. // your code goes here
  33. int t;
  34. cin>>t;
  35. while(t--){
  36. solve();
  37. }
  38. return 0;
  39. }
Success #stdin #stdout 0.01s 5284KB
stdin
1
18 886531871815571953
2609 10 5098 9591 949 8485 6385 4586 1064 5412 6564 8460 2245 6552 5089 8353 3803 3764
stdout
-316435155