C. Quests(动态规划)

Problem - C - Codeforces

晚上写的时候用了动态规划

但是规划的不好

很蠢的开了dp[][]

其实开个dp[]就好了

Time limit exceeded on test 3

#include<bits/stdc++.h>
#define int long long
using namespace std;
 
signed main(){
	int t,s,k,maxd,maxn;
	cin>>t;
	while(t--){
		cin>>s>>k;
		vector<vector<int>>dp(k,vector<int>(2));
		vector<int>jya(s);
		vector<int>jyb(s);
		for(int i=0;i<s;i++){
			cin>>jya[i];
		}
		for(int i=0;i<s;i++){
			cin>>jyb[i];
		}
		
		dp[0][0]=jya[0];
		maxd=jyb[0];
		dp[0][1]=dp[0][0]+(k-1)*maxd;
		maxn=dp[0][1];
		for(int i=1;i<s&&i<k;i++){
		dp[i][0]=dp[i-1][0]+jya[i];
		maxd=max(maxd,jyb[i]);
		dp[i][1]=dp[i][0]+(k-1-i)*maxd;
		
		maxn=max(maxn,dp[i][1]);
			
		}
		if(k<=s)
		maxn=max(maxn,dp[k-1][0]);
		
		cout<<maxn<<endl;
		    jya.clear();
		    jya.shrink_to_fit();
		    jyb.clear();
		    jyb.shrink_to_fit();	
			dp.clear();
			dp.shrink_to_fit();
		
	}
}

优化后的代码

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=2*10e5 +9;
int dp[N];
void solve(){
    int n,k,maxd,maxn;
    cin>>n>>k;
    int jya[n+1]={0},jyb[n+1]={0},sum[n+1]={0};
    for(int i=1;i<=n;i++) cin>>jya[i];
    for(int i=1;i<=n;i++) sum[i]=sum[i-1]+jya[i];
    for(int i=1;i<=n;i++) cin>>jyb[i];
    dp[0]=0;
    maxd=jyb[1];
    dp[1]=sum[1]+maxd*(k-1);
    maxn=dp[1];
    for(int i=1;i<=n&&i<=k;i++){
        if(i!=1){
        maxd=max(jyb[i],maxd);
        dp[i]=sum[i]+maxd*(k-i);
        }
    maxn=max(dp[i],maxn);
    }
    cout<<maxn<<endl;
}

signed main(){
	int t;
	cin>>t;
	while(t--)
	solve();
}