diff options
Diffstat (limited to 'rotate_function.cpp')
-rw-r--r-- | rotate_function.cpp | 29 |
1 files changed, 29 insertions, 0 deletions
diff --git a/rotate_function.cpp b/rotate_function.cpp new file mode 100644 index 0000000..d764feb --- /dev/null +++ b/rotate_function.cpp @@ -0,0 +1,29 @@ +#include "leetcode.h" + +class Solution { +public: + int maxRotateFunction(vector<int> &nums) { + long sum = 0, fn = 0; + int len = nums.size(); + for (int i = 0; i < len; i++) { + sum += nums[i]; + fn += (i * nums[i]); + } + long l = 1, r, newFn = fn; + while (l < len) { + r = l + len - 1; + long removed = (l - 1) * nums[l - 1]; + long added = r * nums[r % len]; + newFn -= newFn - removed + added - sum; + fn = max(fn, newFn); + l++; + } + return (int)fn; + } +}; + +int main() { + Solution obj; + vector<int> nums = {4, 3, 2, 6}; + cout << obj.maxRotateFunction(nums); +} |