题意:长度为n的01串(?可以表示0或1),表示成的二进制,转成格雷码,每个位置有一个分值,若格雷码的i位置为1则可以得到val【i】分,求最多能拿多少分。
设dp【i】【0】为i位置为0时的最高分,dp【i】【1】为i位置为1时的最高分 ,然后分类判断一下即可。
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 200010;
int dp[N][2];
int v[N];
char a[N];
int main()
{
int tot;
scanf("%d", &tot);
for( int ca = 1; ca <= tot; ++ca ) {
scanf("%s", a+1);
int n = strlen( a+1 );
a[0] = '0';
for( int i = 1; i <= n; ++i ) {
scanf("%d", &v[i]);
}
memset( dp, 0, sizeof( dp ) );
for( int i = 1; i <= n; ++i ) {
if( a[i] == '0' ) {
if( a[i-1] == '0' )
dp[i][0] = dp[i-1][0];
else if( a[i-1] == '1' )
dp[i][0] = dp[i-1][1] + v[i];
else if( a[i-1] == '?' ) {
dp[i][0] = max( dp[i-1][1] + v[i], dp[i-1][0] );
}
}
else if( a[i] == '1' ) {
if( a[i-1] == '0' )
dp[i][1] = dp[i-1][0] + v[i];
else if( a[i-1] == '1' )
dp[i][1] = dp[i-1][1];
else if( a[i-1] == '?' ) {
dp[i][1] = max( dp[i-1][1], dp[i-1][0] + v[i] );
}
}
else if( a[i] == '?' ) {
if( a[i-1] == '0' ) {
dp[i][0] = dp[i-1][0];
dp[i][1] = max( dp[i-1][0] + v[i], dp[i-1][1] );
}
else if( a[i-1] == '1' ) {
dp[i][1] = dp[i-1][1];
dp[i][0] = max( dp[i-1][1] + v[i], dp[i-1][0] );
}
else if( a[i-1] == '?' ) {
dp[i][0] = max( dp[i-1][1] + v[i], dp[i-1][0] );
dp[i][1] = max( dp[i-1][0] + v[i], dp[i-1][1] );
}
}
}
printf("Case #%d: %d\n", ca, max( dp[n][1], dp[n][0] ) );
}
return 0;
}