1393 0和1相等串 鸽笼原理 || 化简dp公式

http://www.51nod.com/onlineJudge/questionCode.html#!problemId=1393

正解一眼看出来的应该是鸽笼原理。记录每个位置的前缀和，就是dp[i][1]表示前i个数中，1的个数。dp[i][0]同理。

然后计算出每一个位置的dp[i][1] - dp[i][0]，如果和前面的出现相同，那么这一段就是贡献。

也可以从化简dp公式来看。

也是一样的都dp[i][0] && dp[i][1]

那么设区间为be, en

如果要相同，则需要dp[en][1] - dp[be - 1][1] == dp[en][0] - dp[be - 1][0]

化简下，就是上面的式子。

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <assert.h>
#define IOS ios::sync_with_stdio(false)
using namespace std;
#define inf (0x3f3f3f3f)
typedef long long int LL;

#include <iostream>
#include <sstream>
#include <vector>
#include <set>
#include <map>
#include <queue>
#include <string>
#include <bitset>
const int maxn =  + ;
int dp[maxn][];
char str[maxn];
vector<int>pos[maxn * ];
void work() {
    scanf("%s", str + );
    int lenstr = strlen(str + );
    int one = , zero = ;
    for (int i = ; i <= lenstr; ++i) {
        if (str[i] == '') {
            zero++;
            dp[i][] = dp[i - ][] + ;
            dp[i][] = dp[i - ][];
        } else {
            one++;
            dp[i][] = dp[i - ][] + ;
            dp[i][] = dp[i - ][];
        }
    }
    for (int i = ; i <= lenstr; ++i) {
        pos[dp[i][] - dp[i][] + lenstr].push_back(i);
    }
    int ans = ;
    for (int i = ; i <=  * lenstr; ++i) {
        if (pos[i].size() <= ) continue;
        ans = max(ans, pos[i].back() - pos[i][]);
    }
    cout << ans << endl;
    return;
}

int main() {
#ifdef local
    freopen("data.txt", "r", stdin);
//    freopen("data.txt", "w", stdout);
#endif
    work();
    return ;
}