1498 D. Bananas in a Microwave(dp转移的优化)

LINK

定义$f[i][j]$表示前$i$个操作是否能到状态$j$

枚举状态$O(nm)$,转移的时候需要枚举$a_i$操作次数,复杂度$O(m)$

整体复杂度是$O(n{m}^2)$，无法通过

考虑优化

当我们枚举到第$i$次操作,此时$t_i=2,x_i,y_i$,枚举的状态是$f[i-1][k]$

尝试转移到$f[i][k],f[i][k\ast x_i],f[i][k\ast x_i^2]...f[i][k\ast x_i^{y_i}]$

这中间是不是存在很多重复??比如下次我枚举的状态是$f[i-1][k\ast x_i^2]$

那么我会去尝试转移到$f[i][k\ast x_i^2],f[i][k\ast x_i^3]...f[i][k\ast x_i^{y_i+2}]$

所以这个枚举过程是可以提前停下来的。如果发现后面枚举的状态可以更新自己能更新的状态,那么自己就不需要更新了,可以$break$掉

换句话说,当我们尝试用$f[i-1][k]$转移到$f[i][k],f[i][k\ast x_i]$时,没有任何问题

当我们尝试用$f[i-1][k]$转移到$f[i][k\ast x_i^2]$时,就应该break掉了!!

因为存在$f[i-1][k\ast x_i^2]$为真,这个状态更新的范围一定覆盖了$f[i-1][k]$所能更新的范围

代码比较好写。

时间复杂度是严格的$O(nm)$,因为每个状态至多更新一次

一点细节

向上取整用小数+ceil函数似乎存在精度问题,自己模拟一下就好了

注意开long long

#include <bits/stdc++.h>
using namespace std;
#define int long long
const int maxn = 2e5+10;
bool f[2][maxn];
int x[maxn],t[maxn],y[maxn],ans[maxn],n,m;
void update(int &cur,int i)
{
	if( t[i]==1 )	cur = cur + ( x[i]+99999 )/100000;
	else	cur = ( cur*x[i]+99999 )/100000;
}
signed main()
{
	memset( ans,-1,sizeof ans );
	cin >> n >> m;
	for(int i=1;i<=n;i++)	cin >> t[i] >> x[i] >> y[i];
	
	f[0][0] = true;
	for(int i=1;i<=n;i++)
	{
		int u = i&1, v = u^1;
		for(int j=0;j<=m;j++)	f[u][j] = f[v][j];//继承上一轮的状态 
		for(int k=0;k<=m;k++)
		{
			if( !f[v][k] )	continue;//上一次根本没完成 
			int cur = k;
			for(int w=1;w<=y[i];w++)
			{
				update( cur,i );
				if( cur>m || f[v][cur] )	break;
				ans[cur] = i; f[u][cur] = true;
			}
		}
	}
	
	for(int i=1;i<=m;i++)	cout << ans[i] << " ";
	return 0;
}

怕有人看不懂滚动数组的,正常的也写了一份,都差不多

#include <bits/stdc++.h>
using namespace std;
#define int long long
const int maxn = 2e5+10;
bool f[209][maxn];
int x[maxn],t[maxn],y[maxn],ans[maxn],n,m;
void update(int &cur,int i)
{
	if( t[i]==1 )	cur = cur + ( x[i]+99999 )/100000;
	else	cur = ( cur*x[i]+99999 )/100000;
}
signed main()
{
	memset( ans,-1,sizeof ans );
	cin >> n >> m;
	for(int i=1;i<=n;i++)	cin >> t[i] >> x[i] >> y[i];
	
	f[0][0] = true;
	for(int i=1;i<=n;i++)
	for(int k=0;k<=m;k++)
	{
		f[i][k] |= f[i-1][k];
		if( !f[i-1][k] )	continue;//上一次根本没完成 
		int cur = k;
		for(int w=1;w<=y[i];w++)
		{
			update( cur,i );
			if( cur>m || f[i-1][cur] )	break;
			ans[cur] = i; f[i][cur] = true;
		}
	}
	
	for(int i=1;i<=m;i++)	cout << ans[i] << " ";
	return 0;
}