P4933 大师--二维转移dp

在这里插入图片描述
好家伙第一次写50行代码没看题解没调试就ac了,hh,主要是能在草纸上准确的把样例跑一遍再开始写;
P4933 大师
洛谷的一道题,然后状态转移就是

if (nums[i] - nums[j] == pair1.first||pair1.first==Null)
				{
					dp[i].push_back({ nums[i] - nums[j],pair1.second });
					//这里只做状态转移不做状态累积
				}

这里面只做状态转移是因为我用的是二维的dp,就是双重循环的dp,如果是一维线性的话是累计过去的,不过大量的状态转移需要优化一下,就是有冗余状态的出现:
代码:

sort(dp[i].begin(), dp[i].end(), [](dz &a1, dz &a2) {return a1.first < a2.first; });
		int j = 1;
		while (j != dp[i].size())
		{
			if (dp[i][j].first == dp[i][j - 1].first)
			{
				dp[i][j - 1].second = ((ll)dp[i][j - 1].second + dp[i][j].second) % mod;
				dp[i].erase(dp[i].begin() + j);
			}
			else
				j++;
		}

这块是做一步优化,就是合并状态先sort一下比较好合,没有二维的数组是因为那个等差数列的差有负数

#include<bits/stdc++.h>
#define Null INT_MAX
using namespace std;
typedef pair<int, int> dz;
typedef long long ll;
const int length = 1e3;
int nums[length];
int mod = 998244353;
vector<vector<pair<int,int>>> dp(length);
int main(void)
{
	int n;
	scanf_s("%d", &n);
	for (int i = 0; i < n; i++)
	{
		scanf_s("%d", &nums[i]);
		dp[i].push_back({ Null,1 });
	}
	for (int i = 1; i < n; i++)
	{
		for (int j = 0; j < i; j++)
		{
			for (int k = 0; k < dp[j].size(); k++)
			{
				pair<int, int> pair1 = dp[j][k];
				if (nums[i] - nums[j] == pair1.first||pair1.first==Null)
				{
					dp[i].push_back({ nums[i] - nums[j],pair1.second });
					//这里只做状态转移不做状态累积
				}
			}
		}
		//弄完之后在这优化一下,时间复杂度大概是O(nlgn)
		sort(dp[i].begin(), dp[i].end(), [](dz &a1, dz &a2) {return a1.first < a2.first; });
		int j = 1;
		while (j != dp[i].size())
		{
			if (dp[i][j].first == dp[i][j - 1].first)
			{
				dp[i][j - 1].second = ((ll)dp[i][j - 1].second + dp[i][j].second) % mod;
				dp[i].erase(dp[i].begin() + j);
			}
			else
				j++;
		}
	}
	int sum = 0;
	for (int i = 0; i < n; i++)
	{
		for (int j = 0; j < dp[i].size(); j++)
		{
			sum = ((ll)sum + dp[i][j].second)%mod;
		}
	}
	printf("%d", sum);
	
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值