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;
}
相关推荐
LaoZhangGong12319 分钟前
使用常数指针作为函数参数
c++·经验分享
边疆.19 分钟前
C++类和对象 (中)
c语言·开发语言·c++·算法
随便取个六字2 小时前
C++学习:类和对象(二)
c++·学习
jimmy.hua3 小时前
C++刷怪笼(9)继承
开发语言·c++
OMGmyhair3 小时前
【 C++ 】C++11的初步学习
开发语言·c++·学习
秋说4 小时前
【数据结构 | PTA】懂蛇语
数据结构·c++
凯子坚持 c5 小时前
String的长度有限,而我对你的思念却无限延伸
c++
何曾参静谧5 小时前
「C/C++」C++20 之 #include<ranges> 范围
c语言·c++·c++20
挨代码6 小时前
C++ —— 常见的初始化
c++
酒鬼猿6 小时前
C++初阶(七)--类和对象(4)
开发语言·c++