CF The Walkway

Problem - 1858B - Codeforces

大致题意：

一条路上有几个特殊点，一个人会在以下这几种情况吃一块饼干

• 经过特殊点
• 在1点
• 已经经过k段路程没吃饼干

现在可以删除一个特殊点，问这个人最少吃多少饼干以及有多少种删除方式

思路：先预处理出每个区间（左开右闭）需要吃多少个饼干，用pre前缀和数组统计前i个特殊点区间吃多少个饼干，枚举第i个特殊点删掉后需要吃多少块饼干，例如我们要删掉第i个特殊点，删点之后的饼干数=[1，n]整个区间需要吃的饼干数-（pre[i+1]-pre[i-1]-1)+(v[i+1]-v[i-1])/m-((v[i+1]-v[i-1])%m==0)，（为什么pre[i+1]-pre[i-1]-1需要减一，因为我们要计算第i个点对区间（v[i-1],v[i+1])的影响，不能包含第i+1个特殊点，所以减去的是第i+1个特殊点。）

AC Code:

void solve()
{

 int t;cin>>t;

 while(t--)
 {
    int n,k,d;cin>>n>>k>>d;
    vector<int> dp(k+5,0),v(k+5,0),pre(k+5,0);
   pre[0]=dp[0]=1;v[0]=1;
    int ans=1;
    for(int i=1;i<=k;i++)
    {
       cin>>v[i];
       dp[i]=(v[i]-v[i-1])/d+1-((v[i]-v[i-1])%d==0);
       pre[i]=pre[i-1]+dp[i];//前缀和计算前i个区间需要吃多少个饼干
       ans+=dp[i];
    }
    dp[k+1]=(n-v[k])/d;
    pre[k+1]=pre[k]+dp[k+1];
    map<int,int> mp;
     ans+=dp[k+1];
   //  cout<<ans<<endl;
     int temp=ans;
     v[k+1]=n;
    for(int i=1;i<k;i++)
    {
      int cnt=(v[i+1]-v[i-1])/d-((v[i+1]-v[i-1])%d==0);//删掉第i个点时，区间是左开右开
                                                       //因为不能删掉第i-1和第i+1个特殊点
      int cnt1=pre[i+1]-pre[i-1];
     //   cout<<cnt<<' '<<cnt1<<endl;
      ans=min(ans,temp-cnt1+cnt+1);
      mp[temp-cnt1+cnt+1]++;
    }
       
         int cnt=(v[k+1]-v[k-1])/d;//删掉第k个特殊点时，区间是左开右闭的，
                                   // 因为v[k+1]=n一定得包含在这个区间里
         int cnt1=pre[k+1]-pre[k-1];
    // cout<<cnt<<' '<<cnt1<<endl;
         ans=min(ans,temp-cnt1+cnt);
         mp[temp-cnt1+cnt]++;

         cout<<ans<<' '<<mp[ans]<<endl;
 } 
}