Codeforces 883D. Packmen Strike Back(二分+DP)
题意:一条长为n线上有几个吃豆人和几个豆子,你可以控制吃豆人的移动方向,选定方向之后吃豆人便会一直向这个方向走,问能吃到的最大豆子数和为了达到这个目标所需的最少时间。
思路:最大的豆子数量其实就是所有的豆子,唯一有个特例就是只有一个人,有两个人以上的话,不难想出只要两人面对面走就能吃到所有豆子了,所以先处理只有一个人的情况。
void straight()
{
int sta=pos[0];
int sumlef=0,timlef=0,k=0,sumrig=0,timrig=0;
for(int i=sta-1;i>=1;i--)
{
k++;
if(s[i]=='*')
{
sumlef++;
timlef=k;
}
}
k=0;
for(int i=sta+1;i<=n;i++)
{
k++;
if(s[i]=='*')
{
sumrig++;
timrig=k;
}
}
if(sumlef<sumrig || (sumlef==sumrig && timrig<timlef))
cout<<sumrig<<" "<<timrig<<endl;
else
cout<<sumlef<<" "<<timlef<<endl;
}
然后就是怎么求最小时间了,数据范围是1e6,所以应该是O(nlogn)或O(n)的算法,再加上这道题明显符合单调性(X秒内能吃到Y颗豆子,那么X+1秒内一定可以吃到Y颗豆子)所以我们可以二分时间。
void binary()
{
int l=0,r=n,mid,ans=0;
while(l<=r)
{
mid=(l+r)/2;
if(check(mid))
{
r=mid-1;
ans=mid;
}
else
l=mid+1;
}
cout<<sum[n]<<" "<<ans<<endl;
}
然后就是如何check的问题,我们可以用DP来解决,dp[i]表示前i个人能吃到的从左数最右边的豆子(就是前i个人能吃到的最右边的豆子),dp[i]可以从dp[i-1]再加上第i个人吃的豆子转移得到,但一共有三种情况,取最大值。
一、第i个人向左走。
二、第i个人向右走。
三、第i-1个人向右,第i个人向左。
分别算就行了,当然,为了求第i个人到第i-1个人之见豆子的数量,要提前预处理好前缀和。
void prework()
{
for(int i=1;i<=n;i++)
{
if(s[i]=='*')
sum[i]=sum[i-1]+1;
else
sum[i]=sum[i-1];
if(s[i]=='P')
pos.pb(i);
}
m=pos.size();
}
inline bool nothing(int l,int r)
{
return (r<l || !(sum[r]-sum[l-1]));
}
bool check(int tim)
{
memset(dp,0,sizeof(dp));
for(int i=0;i<m;i++)
{
if(nothing(dp[i]+1,pos[i]-tim-1))
dp[i+1]=max(dp[i+1],pos[i]);
if(nothing(dp[i]+1,pos[i]-1))
dp[i+1]=max(dp[i+1],pos[i]+tim);
if(i>=1 && nothing(dp[i-1]+1,pos[i]-tim-1) && pos[i]-tim<pos[i-1])
dp[i+1]=max(dp[i+1],pos[i-1]+tim);
}
if(nothing(dp[m]+1,n))
return 1;
return 0;
}
最后是完整的AC代码
//头文件日常省略
using namespace std;
const int maxn=1000005;
string s;
int n,m;
vector<int> pos;
int sum[maxn],dp[maxn];
void prework()
{
for(int i=1;i<=n;i++)
{
if(s[i]=='*')
sum[i]=sum[i-1]+1;
else
sum[i]=sum[i-1];
if(s[i]=='P')
pos.pb(i);
}
m=pos.size();
}
void straight()
{
int sta=pos[0];
int sumlef=0,timlef=0,k=0,sumrig=0,timrig=0;
for(int i=sta-1;i>=1;i--)
{
k++;
if(s[i]=='*')
{
sumlef++;
timlef=k;
}
}
k=0;
for(int i=sta+1;i<=n;i++)
{
k++;
if(s[i]=='*')
{
sumrig++;
timrig=k;
}
}
if(sumlef<sumrig || (sumlef==sumrig && timrig<timlef))
cout<<sumrig<<" "<<timrig<<endl;
else
cout<<sumlef<<" "<<timlef<<endl;
}
inline bool nothing(int l,int r)
{
return (r<l || !(sum[r]-sum[l-1]));
}
bool check(int tim)
{
memset(dp,0,sizeof(dp));
for(int i=0;i<m;i++)
{
if(nothing(dp[i]+1,pos[i]-tim-1))
dp[i+1]=max(dp[i+1],pos[i]);
if(nothing(dp[i]+1,pos[i]-1))
dp[i+1]=max(dp[i+1],pos[i]+tim);
if(i>=1 && nothing(dp[i-1]+1,pos[i]-tim-1) && pos[i]-tim<pos[i-1])
dp[i+1]=max(dp[i+1],pos[i-1]+tim);
}
if(nothing(dp[m]+1,n))
return 1;
return 0;
}
void binary()
{
int l=0,r=n,mid,ans=0;
while(l<=r)
{
mid=(l+r)/2;
if(check(mid))
{
r=mid-1;
ans=mid;
}
else
l=mid+1;
}
cout<<sum[n]<<" "<<ans<<endl;
}
int main()
{
cin>>n>>s;
s=" "+s;
prework();
if(m==1)
straight();
else
binary();
return 0;
}