2023大厂真题提交网址(含题解):
www.CodeFun2000.com(http://101.43.147.120/)
最近我们一直在将收集到的机试真题制作数据并搬运到自己的OJ上,供大家免费练习,体会真题难度。现在OJ已录入50+道2023年最新大厂真题,同时在不断的更新。同时,可以关注"塔子哥学算法"公众号获得每道题的题解。
前言:
学习到一种新的优化 d p dp dp的技巧:水平线优化。
题目大意:略
题目思路:dp,前缀和优化,水平线优化。
考虑朴素的动态规划: d p ( i , j ) dp(i,j) dp(i,j)代表填完前 i i i位且前缀和为 j j j的合法方案数.
要么符合操作一,第一种转移: d p ( i , j ) ← d p ( i − 1 , j − b i ) dp(i,j)\leftarrow dp(i-1,j-b_i) dp(i,j)←dp(i−1,j−bi)
要么符合操作二,第二种转移: d p ( i , b i ) ← ∑ − i n f i n f d p ( i − 1 , j ) − d p ( i − 1 , 0 ) dp(i,b_i)\leftarrow \sum_{-inf}^{\ \ \ inf}dp(i-1,j)-dp(i-1,0) dp(i,bi)←∑−inf infdp(i−1,j)−dp(i−1,0)
注:当 j = b i j=b_i j=bi时,同时满足操作一,操作二。所以得减去 d p ( i − 1 , 0 ) dp(i-1,0) dp(i−1,0).
不难发现,第一种转移就是将整个 d p dp dp数组右移 b i b_i bi.那么我们维护一个 o f f off off变量代表偏移量。每次 o f f − = b i off -= b_i off−=bi. 这样我们优化掉第一维.
对于第二种转移,我们维护一个 s s s代表 d p ( i − 1 ) dp(i-1) dp(i−1)的总和。
然后 d p ( b i + o f f ) = s − d p ( o f f + b i ) dp(b_i+off)=s-dp(off+b_i) dp(bi+off)=s−dp(off+bi) ---- 因为访问 d p ( o f f ) dp(off) dp(off)等价于访问 d p ( i , 0 ) dp(i,0) dp(i,0),那么访问 d p ( o f f + b i ) dp(off+b_i) dp(off+bi)等价于访问 d p ( i − 1 , 0 ) dp(i-1,0) dp(i−1,0)
时间复杂度计算:
根据上面的转移我们不难发现,每次多出一个 b i b_i bi,我们前缀和的总数至多增加一个.我们用map实现dp数组。时间复杂度为: O ( n l o g n ) O(nlogn) O(nlogn)
AC代码:
#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pii pair<int,int>
#define pb push_back
#define mp make_pair
const int maxn = 1e6 + 5;
const int mod = 1e9 + 7;
int a[maxn];
int main()
{
ios::sync_with_stdio(false);
int t; cin >> t;
while (t--){
int n ; cin >> n;
for (int i = 1 ; i <= n ; i++) cin >> a[i];
map<ll,ll> dp;
dp[0] = 1;
ll s = 1 , off = 0 , add;
for (int i = 1 ; i <= n ; i++){
off -= a[i];
add = (s - dp[off + a[i]] + mod) %mod;
dp[a[i] + off] = (dp[a[i] + off] + add) % mod;
s = (s + add) % mod;
}
cout << s << endl;
}
return 0;
}
参考博客:
https://blog.csdn.net/Fighting_Peter/article/details/113802017