请帮我把这段代码改成 O(n) 的

#include<bits/stdc++.h>
using namespace std;
long long dp[1000010];
long long f[1000010],d[1000010],s[1000010];
int main()
{
    ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
    int n,k;
    cin>>n>>k;
    for(int i=1;i<=n;i++) cin>>f[i];
    for(int i=1;i<=n;i++)
    {
        cin>>d[i];
        s[i]=s[i-1]+d[i];
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=i-1;j>=1;j--)
        {
            if(f[i]==f[j])
            {
                dp[i]=max(dp[i],dp[j-1]+s[i]-s[j-1]);
            }
        }
    }
    cout<<dp[n];
    return 0;
}