http://acm.csust.edu.cn/contest/152/problem/G
思路:
如果按照枚举最后一段的长度去做,复杂度将到达O(n^2*k)的。无法接受。
但是观察到gcd很小只有100。
所以将gcd作为一维
dp[i][j][k] 表示第i个数,当前部分的gcd为j,已经分到第k部分的答案
对于第i个,要不和前面的合成一段。要不分开。
对于分开:dp[i][a[i]][k] = max(dp[i][a[i]][k], dp[i-1][j][k-1] + a[i]); 即开出新的一段
对于合并,其对应含义就是在前i-1中找最后一段,也就是第k段和i合并,对于这个前i-1中的最后一段gcd是多少,也是枚举的。然后去掉原来这第k段的贡献并且更新为现在的。
李聚聚的顺推图更好理解
#include<iostream>
#include<vector>
#include<queue>
#include<cstring>
#include<cmath>
#include<map>
#include<set>
#include<cstdio>
#include<algorithm>
#define debug(a) cout<<#a<<"="<<a<<endl;
using namespace std;
const int maxn=1e4+10;
typedef int LL;
inline LL read(){LL x=0,f=1;char ch=getchar(); while (!isdigit(ch)){if (ch=='-') f=-1;ch=getchar();}while (isdigit(ch)){x=x*10+ch-48;ch=getchar();}
return x*f;}
LL dp[maxn][110][51];
LL a[maxn];
int main(void){
cin.tie(0);std::ios::sync_with_stdio(false);
LL n,m;cin>>n>>m;
for(LL i=1;i<=n;i++) cin>>a[i];
memset(dp,-0x3f,sizeof(dp));
dp[1][a[1]][1]=a[1];
for(LL i=1;i<=n;i++){
for(LL j=1;j<=100;j++){
for(LL k=1;k<=m;k++){
LL G=__gcd(a[i],j);
///不合并
dp[i][a[i]][k]=max(dp[i][a[i]][k],dp[i-1][j][k-1]+a[i]);
///合并
dp[i][G][k]=max(dp[i][G][k],dp[i-1][j][k]-j+G);
}
}
}
LL ans=0;
for(LL j=1;j<=100;j++){
ans=max(ans,dp[n][j][m]);
}
cout<<ans<<"\n";
return 0;
}
但是光光说可以复杂度O(nk)做...没看懂
先贴代码
/*** keep hungry and calm CoolGuang! ***/
//#pragma GCC optimize(3)
#include <bits/stdc++.h>
#define debug(x) cout<<#x<<":"<<x<<endl;
#define dl(x) printf("%lld\n",x);
#define di(x) printf("%d\n",x);
typedef long long ll;
typedef unsigned long long ull;
using namespace std;
const ll INF= 3e18+7;
const ll maxn = 1e4+700;
const int mod= 998244353;
const double eps = 1e-9;
const double PI = acos(-1);
template<typename T>inline void read(T &a){char c=getchar();T x=0,f=1;while(!isdigit(c)){if(c=='-')f=-1;c=getchar();}
while(isdigit(c)){x=(x<<1)+(x<<3)+c-'0';c=getchar();}a=f*x;}
ll n,m,p;
int a[maxn];
int mx[maxn][55],cop[maxn];
int dp[maxn][55];
int main(){
read(n);read(m);
for(int i=1;i<=n;i++) read(a[i]);
int cnt = 0,tcnt = 0;
for(int i=1;i<=n;i++){
for(int j=1;j<=cnt;j++) cop[j] = __gcd(a[i],cop[j]);
cop[++cnt] = a[i];
for(int k=0;k<=m;k++) mx[cnt][k] = dp[i-1][k];
tcnt = cnt;cnt = 0;
for(int k=1;k<=tcnt;k++){
cop[++cnt] = cop[k];
for(int j=0;j<=m;j++) mx[cnt][j] = mx[k][j];
int s = k+1;
while(s <= tcnt && cop[k] == cop[s]){
for(int j=0;j<=m;j++) mx[cnt][j] = max(mx[cnt][j],mx[s][j]);
s++;
}
k = s-1;
}
for(int k=1;k<=cnt;k++){
for(int j=1;j<=m;j++){
dp[i][j] = max(dp[i][j],mx[k][j-1]+cop[k]);
}
}
dp[i][0] = -mod;
}
di(dp[n][m]);
return 0;
}
/**
**/