396. 旋转函数
递推,这种题一般就是手写样例,找关联性,往往不会很难。
一般地关联性怎么找呢?就是看上一个状态和下一个状态的关联,通常总结成递推或者是整理出一个通项公式。递推经常会用到最大最小值、前缀和、贪心的思想。
本题可以观察到,f[i+1]
和f[i]
区别在于除了某一项之外,其它项累加了一次,同时某一项减去size(nums)*k
就可以消除那一项的影响,因此可以得到递推公式惹。
// shiran
#include <bits/stdc++.h>
using namespace std;
#define rep(i, a, n) for (int i = a; i < n; i++)
#define per(i, n, a) for (int i = n - 1; i >= a; i--)
#define sz(x) (int)size(x)
#define fi first
#define se second
#define all(x) x.begin(), x.end()
#define pb push_back
typedef long long ll;
typedef pair<int, int> PII;
const int mod = 1000000007;
const int N = 1010;
int dx[4] = {-1, 0, 1, 0}, dy[4] = {0, 1, 0, -1};
class Solution
{
public:
int maxRotateFunction(vector<int> &nums)
{
int n = sz(nums);
int sum = accumulate(all(nums), 0);
int f = 0, maxn = 0;
rep(i, 0, n)
f += i * nums[i];
maxn = f;
per(i, n, 1)
{
f = f + sum - n * nums[i],
maxn = max(maxn, f);
}
return maxn;
}
};