题目
给定a,b(1<=a,b<=1e5),素数p=1e5+3,
给定长度n(n<=5e3)的两个数组,va[]和vb[](1<=vai,vbj<=1e5)
要求把va和vb代入上式中求出wa[]和wb[]
对于wa[]、wb[]数组,将两个数组归得到最后完整数组c[]
c中每一段长度为k的连续的全都相同的值v,对答案的贡献是(k-1)*v
求最后最大的贡献的和
思路来源
自己瞎莽+梁神思路
题解
欧拉降幂,用递归把va[]和vb[]搞成wa[]和wb[],
然后考虑dp[5000][5000][2],dp[i][j][op]代表当前c数组已经选了i个wa,j个wb
op=0代表这一次选i,op=1代表这一次选j
由于这题还会卡内存,所以滚动数组一下,dp[2][5000][2]过了
代码
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long ll;
const int p=1e5+3;
const int maxn=1e5+10;
const int N=1e5+10;
const int M=5e3+5;
bool ok[maxn];
int prime[maxn],phi[maxn],cnt;
int a,b,n;
int u[N],v[N];
int dp[2][M][2];
void sieve()
{
phi[1]=1;
for(ll i=2;i<maxn;++i)
{
if(!ok[i])
{
prime[cnt++]=i;
phi[i]=i-1;
}
for(int j=0;j<cnt;++j)
{
if(i*prime[j]>=maxn)break;
ok[i*prime[j]]=1;
if(i%prime[j]==0)
{
phi[i*prime[j]]=phi[i]*prime[j];//prime[j]是i的因子 prime[j]的素因子项包含在i的素因子项里
break;
}
else phi[i*prime[j]]=phi[i]*(prime[j]-1);//prime[j]与i互质 phi[i*prime[j]=phi[i]*phi[prime[j]]
}
}
}
int modpow(int x,int n,int mod)
{
int ans=1;
for(;n;n/=2,x=1ll*x*x%mod)
if(n&1)ans=1ll*ans*x%mod;
return ans;
}
int f(int num,int mod)//要求b^b^b..(num个) %mod
{
if(mod==1)return 0;
if(num==1)return b%mod;
return modpow(b,f(num-1,phi[mod])+phi[mod],mod);
}
int main()
{
sieve();
while(~scanf("%d%d",&a,&b))
{
scanf("%d",&n);
for(int i=1;i<=n;++i)
{
scanf("%d",&u[i]);
u[i]=modpow(a,f(u[i],phi[p])+phi[p],p);
}
for(int i=1;i<=n;++i)
{
scanf("%d",&v[i]);
v[i]=modpow(a,f(v[i],phi[p])+phi[p],p);
}
memset(dp,0,sizeof dp);
u[0]=v[0]=p+1;//保证和第二项不同
for(int k=0;k<=n;++k)
{
int i=k&1;
for(int j=0;j<=n;++j)
{
dp[i][j][0]=0;
dp[i][j][1]=0;
}
for(int j=0;j<=n;++j)
{
if(k>0)
{
dp[i][j][0]=max(dp[i][j][0],dp[i^1][j][0]+u[k]*(u[k]==u[k-1]));
dp[i][j][0]=max(dp[i][j][0],dp[i^1][j][1]+u[k]*(u[k]==v[j]));
}
if(j>0)
{
dp[i][j][1]=max(dp[i][j][1],dp[i][j-1][0]+v[j]*(v[j]==u[k]));
dp[i][j][1]=max(dp[i][j][1],dp[i][j-1][1]+v[j]*(v[j]==v[j-1]));
}
}
}
printf("%d\n",max(dp[n&1][n][0],dp[n&1][n][1]));
}
return 0;
}