CF889E Mod Mod Mod 题解

题目传送门 洛谷题目传送门

思路

首先可以想到，$x=a_i-1$ 中肯定是存在最优解的。那么考虑 dp。

设 $d{p}_{i,j}$ 表示第 $i$ 个数，$x$ 取 $j$ 的最大值，分两种情况：

1. $j<a_i$：$d{p}_{i,j}=d{p}_{i-1,j}$，即可以直接转移；
2. $j\ge a_i$：分两种转移：
   • 把 $x$ 变为 $a_x-1$，则 $d{p}_{i,a_i-1}=d{p}_{i-1,j}+(i-1)\times (\lfloor \frac{j+1}{a_i}\rfloor \times a_i-a_i)$；
   • 就取当前的值，则 $d{p}_{i,j\mathrm{mod}{a}_i}=d{p}_{i,j}+(i-1)\times (j-j\mathrm{mod}{a}_i)$。

由于值域比较大，所以直接开个 map 来维护 $dp$ 数组，再继续优化空间，用滚动数组的思想。时间复杂度为 $O(n\log n\log V)$（$V$ 为值域）。

代码

#include <bits/stdc++.h>
#define int long long
#define x first
#define y second
using namespace std;
const int N = 2e5+5;
int n,a[N],ans;
map<int,int> dp;
signed main()
{
//	freopen(".in","r",stdin);
//	freopen(".out","w",stdout);
	ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
	cin>>n;
	for(int i = 1;i<=n;i++)
		cin>>a[i];
	dp[a[1]-1] = 0;
	for(int i = 2;i<=n;i++)
	{
		for(auto j = dp.lower_bound(a[i]);j!=dp.end();dp.erase(j++))
		{
			int x = j->x,y = j->y;
			dp[x%a[i]] = max(dp[x%a[i]],y+(i-1)*(x-x%a[i]));
			dp[a[i]-1] = max(dp[a[i]-1],y+(i-1)*((x+1)/a[i]*a[i]-a[i]));
		}
	}
	for(auto i = dp.begin();i!=dp.end();i++)
	{
		int x = i->x,y = i->y;
		ans = max(ans,x*n+y);
	}
	cout<<ans;
	return 0;
}