fork(9) download
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3.  
  4. int main() {
  5. // your code goes here
  6. int n,m,sum,x;
  7. cin>>n>>m;
  8. set<int> s;
  9. cin>>x;
  10. s.insert(x);
  11. sum=n+1;
  12. m--;
  13. cout<<sum<<"\n";
  14. while(m--)
  15. {
  16. cin>>x;
  17. s.insert(x);
  18. auto pos=s.find(x);
  19. auto pre=pos;
  20. auto nex=pos;
  21. pre--;
  22. nex++;
  23. if(pos!=s.begin()&&pos!=--s.end())
  24. sum+=*(pos)+*(pos);
  25. else if(pos!=s.begin())
  26. sum+=*(pos)+*(pre);
  27. else
  28. sum+=*(pos)+*(nex);
  29. cout<<sum<<"\n";
  30. }
  31. return 0;
  32. }
Success #stdin #stdout 0s 15240KB
stdin
4 4
1 2 3 4
stdout
5
8
13
20