fork download
  1. #include<iostream>
  2. #include<cmath>
  3. #include<algorithm>
  4. using namespace std;
  5. int main()
  6. {
  7.  
  8. long long int answer=0;
  9. long long n;
  10. cin>>n;
  11. long long a[n];
  12. long long totalneg=0;
  13. long long temp;
  14. for(int i=0;i<n;i++)
  15. {
  16. cin>>a[i];
  17.  
  18. }
  19. for(int i=0;i<n;i++)
  20. {
  21. if(a[i]<0)
  22. {
  23. totalneg++;
  24. }
  25. }
  26. long long all;
  27. cin>>all;
  28. if(totalneg==0 || all==0)
  29. {
  30. cout<<answer<<"\n";
  31. return 0;
  32. }
  33. if(totalneg<=all)
  34. {
  35. for(int i=0;i<n;i++)
  36. {
  37. if(a[i]<0)
  38. answer=answer+fabs(a[i]);
  39. }
  40.  
  41. }
  42. else if(totalneg>all )
  43. {
  44. sort(a,a+n);
  45. long long int sub=totalneg-all;
  46. //cout<<sub;
  47. long long int temp1=fabs(a[totalneg-sub]);
  48. long long int p=temp1*all;
  49. // cout<<p<<"dsadasd";
  50. answer=answer+p;
  51. for(int i=0;i<n;i++)
  52. {
  53. a[i]=a[i]+temp1;
  54. // cout<<a[i]<<"sadsada";
  55. }
  56. for(int i=0;i<totalneg;i++)
  57. {
  58. if(a[i]<0)
  59. {
  60. answer=answer+fabs(a[i]);
  61. }
  62. }
  63. }
  64. cout<<answer<<"\n";
  65.  
  66. return 0;
  67. }
Success #stdin #stdout 0s 3348KB
stdin
3
-1 -2 -3
2
stdout
5