#include <iostream> #include <math.h> #include <vector> using namespace std; int howManyBinaryOnesInNumber(int number){ int count = 0; while(number != 0){ number = number & (number - 1); count++; } return count; } int getBiggestAmountOfOnes(int lastAmount, int maxBits){ int biggest = lastAmount + 1; for (int i = lastAmount + 1; i <= maxBits; i++){ if(howManyBinaryOnesInNumber(i) > howManyBinaryOnesInNumber(biggest)){ biggest = i; } } return biggest; } int getSmallestAmountOfOnes(int lastAmount, int localMaxBits){ int smallest = lastAmount + 1; for (int i = lastAmount + 1; i <= localMaxBits; i++){ if(howManyBinaryOnesInNumber(i) < howManyBinaryOnesInNumber(smallest)){ smallest = i; } } return smallest; } int calcMaxQuality(std::vector<int> qualities, int maxBits){ std::vector<int> newQualities; for (int i = 0; i < qualities.size(); i++){ int localMaxBits = maxBits - (qualities.size() - (i + 1)); if(i == 0 && qualities[i] < 0){ newQualities.push_back(0); } else if (i == 0 && qualities[i] >= 0){ newQualities.push_back(getBiggestAmountOfOnes(0, localMaxBits)); } else if (qualities[i] < 0) { newQualities.push_back(getSmallestAmountOfOnes(newQualities[i - 1], localMaxBits)); } else { newQualities.push_back(getBiggestAmountOfOnes(newQualities[i - 1], localMaxBits)); } } int calculatedQuality = 0; for (int i = 0; i <= newQualities.size(); i++){ calculatedQuality += qualities[i] * howManyBinaryOnesInNumber(newQualities[i]); } return calculatedQuality; } int main(){ int howLong, maxBits; std::vector<int> qualities; cin >> howLong >> maxBits; for (int i = 1; i <= howLong; i++){ int quality; cin >> quality; qualities.push_back(quality); } cout << calcMaxQuality(qualities, maxBits); return 0; }
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 | #include <iostream> #include <math.h> #include <vector> using namespace std; int howManyBinaryOnesInNumber(int number){ int count = 0; while(number != 0){ number = number & (number - 1); count++; } return count; } int getBiggestAmountOfOnes(int lastAmount, int maxBits){ int biggest = lastAmount + 1; for (int i = lastAmount + 1; i <= maxBits; i++){ if(howManyBinaryOnesInNumber(i) > howManyBinaryOnesInNumber(biggest)){ biggest = i; } } return biggest; } int getSmallestAmountOfOnes(int lastAmount, int localMaxBits){ int smallest = lastAmount + 1; for (int i = lastAmount + 1; i <= localMaxBits; i++){ if(howManyBinaryOnesInNumber(i) < howManyBinaryOnesInNumber(smallest)){ smallest = i; } } return smallest; } int calcMaxQuality(std::vector<int> qualities, int maxBits){ std::vector<int> newQualities; for (int i = 0; i < qualities.size(); i++){ int localMaxBits = maxBits - (qualities.size() - (i + 1)); if(i == 0 && qualities[i] < 0){ newQualities.push_back(0); } else if (i == 0 && qualities[i] >= 0){ newQualities.push_back(getBiggestAmountOfOnes(0, localMaxBits)); } else if (qualities[i] < 0) { newQualities.push_back(getSmallestAmountOfOnes(newQualities[i - 1], localMaxBits)); } else { newQualities.push_back(getBiggestAmountOfOnes(newQualities[i - 1], localMaxBits)); } } int calculatedQuality = 0; for (int i = 0; i <= newQualities.size(); i++){ calculatedQuality += qualities[i] * howManyBinaryOnesInNumber(newQualities[i]); } return calculatedQuality; } int main(){ int howLong, maxBits; std::vector<int> qualities; cin >> howLong >> maxBits; for (int i = 1; i <= howLong; i++){ int quality; cin >> quality; qualities.push_back(quality); } cout << calcMaxQuality(qualities, maxBits); return 0; } |