好家伙第一次写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);
}