[动态规划]CF – 1005 – D. Polycarp and Div 3

http://codeforces.com/contest/1005/problem/D
定义ans[i]为前缀长度为i的答案，ans[0] = 0，定义f[i]为前缀和模3的最大前缀长度。
那么如何由ans[i – 1]得到ans[i]，我们可以只看前缀长度为i的最后一个字符，就是s[i – 1]，对于这个字符有两种选择，1是放弃，即ans[i] = ans[i – 1]，2是找到前一个前缀和模3相等的最大长度，然后这两个位置之间的部分就相当于一个3的倍数的区间，所以ans[i] = ans[f[sum % 3]] + 1，然后我们把ans更新为这两个值中的最大值就好了，但是要求f[sum % 3]存在。

#include<bits/stdc++.h>
using namespace std;
int f[3] = {-1, -1, -1}, ans[200005];
int main()
{
    string s;
    cin >> s;
    int sum = 0;
    for (int i = 1; i <= s.size(); ++i){
        sum += s[i - 1] - '0';
        ans[i] = ans[i - 1];
        if (f[sum % 3] != -1) ans[i] = max(ans[i], ans[f[sum % 3]] + 1);
        f[sum % 3] = i;
    }
    cout << ans[s.size()] << endl;
    return 0;
}