mirror of
https://gitcode.com/Zengtudor/alg2025.git
synced 2025-12-16 12:13:03 +00:00
feat: 添加P6006.cpp实现三元组计数算法
实现一个动态规划算法来计算满足条件的三元组数量,用于解决特定范围的查询问题。算法通过预处理和动态规划表优化查询效率,并处理了数组边界条件和内存使用问题。
This commit is contained in:
parent
daba2089c4
commit
b2c80a8161
58
src/10/10/P6006.cpp
Normal file
58
src/10/10/P6006.cpp
Normal file
@ -0,0 +1,58 @@
|
|||||||
|
#include <cstdint>
|
||||||
|
#include <iostream>
|
||||||
|
#include <istream>
|
||||||
|
#include <vector>
|
||||||
|
|
||||||
|
using ll = int64_t;
|
||||||
|
|
||||||
|
const ll maxn = 5e3+5;
|
||||||
|
ll n,q, dp[maxn][maxn], a[maxn];
|
||||||
|
|
||||||
|
struct T{
|
||||||
|
const static ll of = 2e6+3;
|
||||||
|
ll v[4*ll(1e6+5)]; // 注意限制的大小要*2
|
||||||
|
inline ll&operator[](const ll n){
|
||||||
|
return v[n+of];
|
||||||
|
}
|
||||||
|
}tot;
|
||||||
|
|
||||||
|
std::vector<ll> used;
|
||||||
|
|
||||||
|
int main(){
|
||||||
|
std::iostream::sync_with_stdio(false);
|
||||||
|
std::cin.tie(nullptr);
|
||||||
|
|
||||||
|
std::cin>>n>>q;
|
||||||
|
for(ll i=1;i<=n;i++){
|
||||||
|
std::cin>>a[i];
|
||||||
|
}
|
||||||
|
|
||||||
|
for(ll i=n-2;i>=1;i--){ // 注意循环顺序!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
|
||||||
|
for(ll j:used)tot[j]=0; // 注意清空耗时
|
||||||
|
used.clear();
|
||||||
|
tot[a[i+1]]++;
|
||||||
|
used.emplace_back(a[i+1]);
|
||||||
|
for(ll j=i+2;j<=n;j++){
|
||||||
|
dp[i][j] = tot[-(a[i]+a[j])];
|
||||||
|
// printf("before dp[%lld][%lld]=%lld\n",i,j,dp[i][j]);
|
||||||
|
dp[i][j] += dp[i+1][j] + dp[i][j-1] - dp[i+1][j-1];
|
||||||
|
tot[a[j]]++;
|
||||||
|
used.emplace_back(a[j]);
|
||||||
|
// printf("after dp[%lld][%lld]=%lld\n",i,j,dp[i][j]);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
while(q--){
|
||||||
|
ll a,b;
|
||||||
|
std::cin>>a>>b;
|
||||||
|
std::cout<<dp[a][b]<<"\n";
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
|
||||||
|
dp[i][j] = i~j的
|
||||||
|
dp[i][j] = dp[i+1][j] + dp[i][j-1] - dp[i+1][j-1]
|
||||||
|
|
||||||
|
|
||||||
|
*/
|
||||||
Loading…
Reference in New Issue
Block a user