写这个题写了1h,赛后无意看见jly神的代码只能膜拜%%%
C. Bouncing Ball
预处理从1→k1\to k1→k开始跳需要添加多少个平台,预处从k+1→nk+1\to nk+1→n这些不难发现由于每次跳k格,只需要利用前缀和思想和前面预处理的结果即可做差求出。
然后枚举删除平台的个数,例用预处理结果直接求答案。
时间复杂度O(n)O(n)O(n)
有一个dt的地方就是如何用数组存预处理结果,我们需要这样大小为数组cnt[k][n/k]
。cnt[i][j]
表示从i
开始向后跳j
次也就是跳了jk
个格子。
数组显然不能开,这里用动态数组vector
存储
#define IO ios::sync_with_stdio(false);cin.tie();cout.tie(0)
#pragma GCC optimize(2)
#include<set>
#include<map>
#include<cmath>
#include<stack>
#include<queue>
#include<random>
#include<bitset>
#include<string>
#include<vector>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<unordered_map>
#include<unordered_set>
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
const int N=100010;int main()
{//IO;int T=1;cin>>T;while(T--){int n,p,k;cin>>n>>p>>k;string s;cin>>s;s="0"+s;ll x,y; cin>>x>>y;ll res=1e18;vector<vector<int> > cnt(k+1,vector<int>(n/k+1,0));for(int i=1;i<=k;i++)for(int j=i;j<=n;j+=k){if(j!=i) cnt[i][(j-i)/k]=cnt[i][(j-i)/k-1];if(s[j]=='0') cnt[i][(j-i)/k]++;}for(int i=1;i<=n;i++){ll now=y*(i-1);int j=i+p-1;if(j>n) continue;int w=(j-1)%k+1;int l=(j-w)/k,r=(n-w)/k;now+=x*cnt[w][r];if(l>0) now-=x*cnt[w][l-1];res=min(res,now);}cout<<res<<'\n';}return 0;
}
jly代码
大体思路就是从后往前计算出需要从某个点向后跳需要添加平台的个数,然后枚举删除平台的个数。
#define IO ios::sync_with_stdio(false);cin.tie();cout.tie(0)
#pragma GCC optimize(2)
#include<set>
#include<map>
#include<cmath>
#include<stack>
#include<queue>
#include<random>
#include<bitset>
#include<string>
#include<vector>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<unordered_map>
#include<unordered_set>
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
const int N=100010;
int main()
{//IO;int T=1;cin>>T;while(T--){int n,p,k;cin>>n>>p>>k;--p;string s;cin>>s;int x,y;cin>>x>>y;int ans=1e9;vector<int> dp(n);for(int i=n-1;i>=0;i--){dp[i]=(s[i]=='0')?x:0;if(i+k<n) dp[i]+=dp[i+k];}for(int i=p;i<n;i++) ans=min(ans,(i-p)*y+dp[i]);cout<<ans<<'\n';}return 0;
}
思维还是不行,别人10分钟A的题,自己1h还差点没过。。
要加油哦~