fork(1) download
copy 
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3.  
  4. bool comp(vector <int> &x, vector <int> y){
  5. 	return x[1]<y[1];
  6. }
  7.  
  8. int findScore(int n, int k, int a[], int b[]){
  9. 	vector< vector <int> > arr;
  10. 	for(int i=0;i<n;i++){
  11. 		vector <int> v;
  12. 		v.push_back(b[i]);
  13. 		v.push_back(a[i]);
  14. 		arr.push_back(v);
  15. 	}
  16. 	sort(arr.begin(), arr.end(), comp);
  17. 	for (int i = 0; i < arr.size(); i++)
  18. 	    {
  19. 	        for (int j = 0; j < arr[i].size(); j++)
  20. 	        {
  21. 	            cout << arr[i][j] << " ";
  22. 	        }   
  23. 	        cout << endl;
  24. 	    }
  25. 	int diversity = 0, experience=0;
  26.  
  27.  
  28. 	for(int i=0;i<k;i++){
  29. 		experience += arr[i][0];
  30. 	}
  31. 	cout<<experience;
  32. 	for(int i=k-1;i>n;i--){
  33. 		// cout<<
  34. 		if(arr[i][1] == arr[i+1][i])
  35. 		{
  36. 			diversity++;
  37. 		}
  38. 	}
  39.  
  40. 	return (diversity*diversity)+experience;
  41. }
  42.  
  43. int main() {
  44. 	// your code goes here
  45. 	int n,k;
  46. 	cin>>n>>k;
  47. 	int a[n], b[n];
  48. 	for(int i=0;i<n;i++){
  49. 		cin>>a[i];
  50. 	}
  51. 	for(int i=0;i<n;i++){
  52. 		cin>>b[i];
  53. 	}
  54.  
  55. 	int result = findScore(n,k,a,b);
  56. 	// cout<<result<<endl;
  57. 	return 0;
  58. }
Success #stdin #stdout 0s 5636KB
comments (?)
stdin
copy 
5
3
1
1
2
2
3
9
7
6
5
1
stdout
copy 
9 1 
7 1 
6 2 
5 2 
1 3 
22
https://ideone.com/T6rg75
language:
C++14 (gcc 8.3)
created:
2 years ago
visibility:
public

Share or Embed source code

Discover > Sphere Engine API

The brand new service which powers Ideone!

Discover > IDE Widget

Widget for compiling and running the source code in a web browser!