LCP 14. 切分数组 dp做法

题目链接

切分数组 

做法：刚开始想成建图跑dij wa了很多发 发现好像不太好做，首先要确定方向，不好确定，边权也不好确定，样例给的 刚开始想的是相邻的i 和  i+1 边权为2  每个nums[i] 和对应的质因数边权为1  然后跑dij  就是dis[n]/2  但是  这个样例：

4

1 3 5 7 

成功的把我hack 了

 

于是换种思路，dp做法，求出每一个nums[i]的最小质因数，维护每个质因数最小能分成几个组，那么dp状态方程为

val是num[i]

while(val>1){
            int t=mi[val];
            f[t]=min(f[t],dp[i-1]);
            dp[i]=min(dp[i],f[t]+1);
            val/=t;
        }

运行代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll gcd(ll a,ll b) { return b?gcd(b,a%b):a;}


int vis[1000010],prime[1000010],len,mi[1000010];
int dp[1000010],f[1000010];

void init(int mx)
{
    for(int i=2;i<=mx;++i){
        if(vis[i]==0){
            prime[++len]=i;
            mi[i]=i;
        }
        for(int j=1;j<=len&&i*prime[j]<=mx;++j){
            vis[i*prime[j]]=1;
            mi[i*prime[j]]=prime[j];
            if(i%prime[j]==0) break;
        }
    }
}
int splitArray(vector<int>& nums) {
    int mx=0;
    for(int v:nums) mx=max(mx,v);

    init(mx);//printf("mx:%d\n",mx);
    int n=nums.size();
    memset(dp,0x3f,sizeof(dp));
    memset(f,0x3f,sizeof(f));
    dp[0]=0;
    for(int i=1;i<=n;++i){
        int val=nums[i-1];
        while(val>1){
            int t=mi[val];
            f[t]=min(f[t],dp[i-1]);
            dp[i]=min(dp[i],f[t]+1);
            val/=t;
        }
    }
    return dp[n];
}
int main()
{


	vector<int>G;
	int n;cin>>n;
	for(int i=1;i<=n;++i){
        int x;
        scanf("%d",&x);
        G.push_back(x);
	}
	cout<<splitArray(G)<<endl;
}

/*
6
2 3 3 2 3 3

4
1 3 5 7
*/

AC代码：

class Solution {
public:
   int vis[1000010],prime[1000010],len,mi[1000010];
int dp[1000010],f[1000010];

void init(int mx)
{
    for(int i=2;i<=mx;++i){
        if(vis[i]==0){
            prime[++len]=i;
            mi[i]=i;
        }
        for(int j=1;j<=len&&i*prime[j]<=mx;++j){
            vis[i*prime[j]]=1;
            mi[i*prime[j]]=prime[j];
            if(i%prime[j]==0) break;
        }
    }
}
int splitArray(vector<int>& nums) {
    int mx=0;
    for(int v:nums) mx=max(mx,v);

    init(mx);//printf("mx:%d\n",mx);
    int n=nums.size();
    memset(dp,0x3f,sizeof(dp));
    memset(f,0x3f,sizeof(f));
    dp[0]=0;
    for(int i=1;i<=n;++i){
        int val=nums[i-1];
        while(val>1){
            int t=mi[val];
            f[t]=min(f[t],dp[i-1]);
            dp[i]=min(dp[i],f[t]+1);
            val/=t;
        }
    }
    return dp[n];
}
};