1090 added.

liuyubobobo · liuyubobobo · commit 74a2f8769b81 · 2019-06-16T04:25:03.000-07:00
diff --git a/1090-Largest-Values-From-Labels/cpp-1090/CMakeLists.txt b/1090-Largest-Values-From-Labels/cpp-1090/CMakeLists.txt
@@ -0,0 +1,6 @@
+cmake_minimum_required(VERSION 3.14)
+project(B)
+
+set(CMAKE_CXX_STANDARD 14)
+
+add_executable(B main.cpp)
diff --git a/1090-Largest-Values-From-Labels/cpp-1090/main.cpp b/1090-Largest-Values-From-Labels/cpp-1090/main.cpp
@@ -0,0 +1,48 @@
+/// Source : https://leetcode.com/problems/largest-values-from-labels/
+/// Author : liuyubobobo
+/// Time   : 2019-06-15
+
+#include <iostream>
+#include <vector>
+#include <unordered_map>
+
+using namespace std;
+
+
+/// Sorting and Greedy
+/// Time Complexity: O(nlogn)
+/// Space Complexity: O(n)
+class Solution {
+public:
+    int largestValsFromLabels(vector<int>& values, vector<int>& labels, int num_wanted, int use_limit) {
+
+        vector<pair<int, int>> vec;
+        for(int i = 0; i < values.size(); i ++)
+            vec.push_back(make_pair(values[i], labels[i]));
+        sort(vec.begin(), vec.end(), greater<pair<int, int>>());
+
+        unordered_map<int, int> freq;
+        int sum = 0, num = 0;
+        for(const pair<int, int>& p: vec)
+            if(freq[p.second] < use_limit){
+
+                sum += p.first;
+                num ++;
+                freq[p.second] ++;
+                if(num == num_wanted)
+                    break;
+            }
+        return sum;
+    }
+};
+
+
+int main() {
+
+    vector<int> values = {5,4,3,2,1};
+    vector<int> labels = {1,1,2,2,3};
+    int num_wanted = 3, use_limit = 1;
+    cout << Solution().largestValsFromLabels(values, labels, num_wanted, use_limit) << endl;
+
+    return 0;
+}
diff --git a/readme.md b/readme.md
@@ -740,4 +740,5 @@ email: [liuyubobobo@gmail.com](mailto:liuyubobobo@gmail.com)
 | 1080 | [Insufficient Nodes in Root to Leaf Paths](https://leetcode.com/problems/insufficient-nodes-in-root-to-leaf-paths/) | [无] | [C++](1080-Insufficient-Nodes-in-Root-to-Leaf-Paths/cpp-1080/) | | |
 | | | | | | |
 | 1089 | [Duplicate Zeros](https://leetcode.com/problems/duplicate-zeros/) | [无] | [C++](1089-Duplicate-Zeros/cpp-1089/) | | |
+| 1090 | [Largest Values From Labels](https://leetcode.com/problems/largest-values-from-labels/) | [无] | [C++](1090-Largest-Values-From-Labels/cpp-1090/) | | |
 | | | | | | |
