C. Gerrymandering 【Codeforces Round 978 (Div. 2)】

C. Gerrymandering

思路:

动态规划

dp[i][j] 表示前i列,末尾是第j种情况下,A的最大得票。

j是第i和i+1列的分布情况,分为0,1,2三种类型:

转移的话只需要搞清楚j=0和j=1的情况即可,j=2与j=1的行是相反的。注意到转移时如果第一行选择横向的连续三格,那么第二行也必须选择横向的三格才能保证不出现漏格。

j=0: 可以转为dp[i+3][0] 、dp[i+1][1] 、dp[i+1][2]

j=1: 可以转为dp[i+2][0] 、dp[I+3][1]

j=2: 可以转为dp[i+2][0] 、dp[I+3][2]

代码:

cpp 复制代码
#include <bits/stdc++.h>
#define endl '\n'
#define int long long 
#define pb push_back
#define pii pair<int,int>
const int MOD = 1e9+7;
const int INF = 0x3f3f3f3f;
typedef long long ll;
using namespace std;
int dp[100005][3];
int c[100005][3];
void solve() {
	memset(dp,-1,sizeof(dp)); //注意初始化dp数组为负
	dp[0][0]=0;
	int n;
	cin>>n;
	for(int j=1;j<=2;j++){
		for(int i=1;i<=n;i++){
			char x;
			cin>>x;
			if(x=='A')c[i][j]=1;
			else c[i][j]=0;
		}
	}
	
	for(int i=0;i<=n;i++){
		for(int j=0;j<=2;j++){
			int val=dp[i][j];
			if(val<0)continue;
			if(j==0){
				int vl=((c[i+1][1]+c[i+2][1]+c[i+3][1])/2 + (c[i+1][2]+c[i+2][2]+c[i+3][2])/2 ) ; //vl是投A的票数
				dp[i+3][0]=max(val+vl,dp[i+3][0]);	
				vl = (c[i+1][1]+c[i+1][2]+c[i+2][1])/2;
				dp[i+1][1]=max(val+vl,dp[i+1][1]);
				vl = (c[i+1][1]+c[i+1][2]+c[i+2][2])/2;
				dp[i+1][2]=max(val+vl,dp[i+1][2]);
			}
			if(j==1){
				int vl=(c[i+1][2]+c[i+2][1]+c[i+2][2])/2;
				dp[i+2][0]=max(val+vl,dp[i+2][0]);
				vl=((c[i+2][1]+c[i+3][1]+c[i+4][1])/2 + (c[i+1][2]+c[i+2][2]+c[i+3][2])/2 );
				dp[i+3][1]=max(val+vl,dp[i+3][1]);	
			}
			if(j==2){
				int vl=(c[i+1][1]+c[i+2][1]+c[i+2][2])/2;
				dp[i+2][0]=max(val+vl,dp[i+2][0]);
				vl=((c[i+2][2]+c[i+3][2]+c[i+4][2])/2 + (c[i+1][1]+c[i+2][1]+c[i+3][1])/2 );
				dp[i+3][2]=max(val+vl,dp[i+3][2]);	
			}
		}
	}
//	
//	for(int i=0;i<=n;i++){
//		for(int j=0;j<3;j++){
//			cout<<"dp["<<i<<"]["<<j<<"]="<<dp[i][j]<<" ";
//		}
//		cout<<endl;
//	}
	cout<<dp[n][0]<<endl;
	
	
}

signed main() {
	cin.tie(0)->ios::sync_with_stdio(0);
	int T = 1;
	cin >> T;
	while (T--) {
		solve();
	}
	return 0;
}
相关推荐
感哥5 分钟前
C++ STL 常用算法
c++
saltymilk11 小时前
C++ 模板参数推导问题小记(模板类的模板构造函数)
c++·模板元编程
感哥11 小时前
C++ lambda 匿名函数
c++
沐怡旸17 小时前
【底层机制】std::unique_ptr 解决的痛点?是什么?如何实现?怎么正确使用?
c++·面试
感哥17 小时前
C++ 内存管理
c++
博笙困了1 天前
AcWing学习——双指针算法
c++·算法
感哥1 天前
C++ 指针和引用
c++
感哥1 天前
C++ 多态
c++
沐怡旸2 天前
【底层机制】std::string 解决的痛点?是什么?怎么实现的?怎么正确用?
c++·面试
River4162 天前
Javer 学 c++(十三):引用篇
c++·后端