DP问题中数组下标通常都是有一定的意义的,但是在C++中如果我们需要用到f[-5][7](随便举的例子)这个数组值时,普通的数组定义是无法实现的。
我们可以用vector<map<int,int>>这种方法实现下标为任意数字的二维数组。
下面我们以一道DP问题来看看具体是怎么实现的。
此题网上有许多种解法。这里使用最朴素的做法。
状态定义:dp[i][j]表示前i个数中总和为j的方案的总数。
状态划分:dp[i][j]中所有的方案可分为两部分,一部分为第i个数取正,则此部分方案总数为dp[i - 1][j - v[i]] 。(v[i]表示第i个数的值)。另一部分为第i个数取负,则此部分方案总数为dp[i -1][j + v[i]]。
则dp[i][j] = dp[i - 1][j - v[i]] + dp[i -1][j + v[i]] (j - v[i]、j + v[i] ∈(-sum,sum))
sum为序列中所有数都为正时的总和。
此题中可见,若使用int dp[][]的定义方法。将无法访问下标为负时的值(j+v[i]、j - v[i]可能为负)。
使用vector<map<int,int>>的方法则可以轻松实现下标为负的二维数组。
#include<bits/stdc++.h>
using namespace std;
int main()
{
int E,temp;
vector<int> v;
scanf("%d", &E);
while(cin >> temp) v.push_back(temp);
int sum = 0;
for(int i = 0;i < v.size();i++) sum += v[i];
vector< map<int,int> > dp(v.size() +1);
for(int i = 0;i <= v.size();i++) //初始化,相当于构建矩阵
for(int j = -sum;j <= sum;j++)
if(i == 0 && j == 0) dp[i][j] = 1;
else dp[i][j] = 0;
for(int i = 1;i <= v.size();i++)
for(int j = -sum;j <= sum; j++){
if(j + v[i - 1] <= sum && j + v[i -1] >= -sum ) dp[i][j] += dp[i - 1][j + v[i -1]];
if(j - v[i - 1] <= sum && j - v[i -1] >= -sum ) dp[i][j] += dp[i - 1][j - v[i -1]];
}
cout << dp[v.size()][E];
return 0;
}
但注意使用这种方法是要遍历数组中的每个元素为其赋值。
这种方法时间复杂度较高,但是思路清晰,简单易懂。各位可以参考一下。(适用于新手)
又发现了一种更简单的方法,本质就是数组指针+偏移量的做法。
转自:数组使用负数下标_Learning Deep Learning-CSDN博客_java 数组下标为负数
# include <bits/stdc++.h>
using namespace std;
const int maxn = 1e6;
///先声明数组
int unused[maxn];
///宏定义方法。这时候a数组就可以为负数了
# define a (unused + (maxn >> 1))
///二维的数组
int x[maxn][10];
# define dp (x + (maxn >> 1))
int main(void)
{
a[-1] = -1;
a[0] = 0;
a[1] = 1;
cout << a[-1] << ' ' << a[0] << ' ' << a[1] << '\n';
dp[-50000][0] = 1234;
cout << dp[-50000][0] << endl;
return 0;
}