C、Save the Magazines
题意:0表示上面没有盖子,1表示有盖子,可以把有盖子的左移一格(第一个严格来说不可以左移),问盖住的最大和
input
4 5 01110 10 5 8 9 6 6 011011 20 10 9 30 20 19 4 0000 100 100 100 100 4 0111 5 4 5 1
output
27 80 0 14
思路:
根据题意比较容易知道是动态规划,dp[i][0]表示到i点时,选择i点最大,dp[i][1]表示将1移动到前面的最大值,转移方程。
为1的话:
dp[i][1]=max(dp[i][1],dp[i-1][1]+a[i-1]); //1只能从前面的dp[i-1][1]中更新
dp[i][0]=max(dp[i-1][0],dp[i-1][1])+a[i];
比较特殊的点,可能就是如果中间有0的话,dp[i][1]也从dp[i-1][0]中更新
比如
011,dp[3][1]=dp[2][1]+a[i-1]
0101,dp[4][1]=dp[3][1]+a[i-1],但是这里的dp[3][1]肯定可以是dp[2][0]
代码:
#include<bits/stdc++.h>
using namespace std;
#define inf 0x3f3f3f3f
#define int long long
#define M 1000005
int T,n;
string s;
int a[M];
int dp[M][2];
signed main(){
cin>>T;
while(T--){
cin>>n;
for(int i=0;i<=n+10;i++) dp[i][0]=dp[i][1]=0;
cin>>s;
for(int i=1;i<=n;i++){
cin>>a[i];
if(s[i-1]=='1'){
dp[i][1]=max(dp[i][1],dp[i-1][1]+a[i-1]);
dp[i][0]=max(dp[i-1][0],dp[i-1][1])+a[i];
}else{
dp[i][1]=max(dp[i-1][1],dp[i-1][0]); //这里要将dp[i-1][0]更新给dp[i][1]
dp[i][0]=dp[i-1][0];
}
}
cout<<max(dp[n][0],dp[n][1])<<endl;
}
return 0;
}
D、Problem with Random Tests
题意:
一个二进制字符串(长度1e6),找两个字串,异或(or)最大,比如11010->11010 or 101 =11111
input
5 11010
output
11111
思路:两个字串,一个肯定是他本身,第二个就要尽量让最前面的0异或为1
最后答案是:
第二个肯定只删除最后的字符,前面保留,所以只要给一个长度d(删除字符数量),就能O(n)一种结果
首先去除字符串的前导0,最前面的就是1
统计前导1的数量l,在统计后面0的数量cnt,(cnt<=l,是如果cnt要>l,cnt就不加了)
三种情况必有一个是答案:
删除数量为:l,l-1,cnt
代码:
#include<bits/stdc++.h>
using namespace std;
#define inf 0x3f3f3f3f
#define int long long
#define M 1000005
int T,n,mx=0,id=0;
string s;
int get(int d){
int mx=0;
for(int i=d;i<=n;i++){
if(s[i]!='1' && s[i-d]!='1')
break;
mx=max(mx,i);
}
return mx;
}
void add(int d){
int cnt=get(d);
if(cnt>mx){
mx=cnt;
id=d;
}
}
signed main(){
cin.tie(nullptr)->sync_with_stdio(false);
cin>>n;
cin>>s;
int f=0;
string v=" ";
for(char c:s){
if(c!='0') f=1;
if(f) v+=c;
}
s=v;
if(s.size()==1) s+="0";
n=s.size()-1;
int l=0;
while(l<n && s[l+1]=='1') l++;
int cnt=0;
while(cnt+l<n && s[cnt+l+1]=='0' && cnt<l) cnt++;
add(cnt); //三种情况
add(l);
add(l-1);
for(int i=1;i<=n;i++){
if(i<=id) cout<<1;
else{
if(s[i]=='1' || s[i-id]=='1') cout<<1;
else cout<<0;
}
}
return 0;
}