Codeforces 383D Antimatter(DP)(题解)

题目链接:Antimatter

题意:给定一个数组,每次可以选取一个连续段,段内每个元素可以取正或负,问有多少种取法可以使得段内和为0(同一段内正负情况不同也为不同方案)

题解:数据就1000个,还告诉我们a的总和<10000,可以想到10^3*10^4=10^7复杂度的DP,我们用dp[i][j]表示以第i个点结尾,总和为j的方案数,枚举ai,算一下ai取正和负对后一个状态的贡献即可,状态转移方程:ai取正时:dp[i][j]+=dp[i-1][j-a[i]],ai取负时:dp[i][j]+=dp[i-1][j+a[i]],因为取负可能导致jK0,我们令初始点base从10000开始即可,

不懂可以看代码

#include<iostream>
#include<stack>
#include<list>
#include<set>
#include<vector>
#include<algorithm>
#include<math.h>
#include<numeric>
#include<map>
#include<cstring>
#include<queue>
#include<iomanip>
#include<cmath>
#include<queue>
#include <bitset>
#include<unordered_map>
	#ifndef local
	#define endl '\n'
#endif */
#define mkp make_pair
using namespace std;
using std::bitset;
typedef long long ll;
typedef long double ld;
const int inf=0x3f3f3f3f;
const ll MAXN=2e5+10;
const ll N=1e3+100;
const ll mod=1e9+7;
const ll hash_p1=1610612741;
const ll hash_p2=805306457;
const ll hash_p3=402653189;
//-----------------------------------------------------------------------------------------------------------------*/
// ll head[MAXN],net[MAXN],to[MAXN],edge[MAXN]/*流量*/,cost[MAXN]//费用;
/* 
void add(ll u,ll v,ll w,ll s){
	to[++cnt]=v;net[cnt]=head[u];edge[cnt]=w;cost[cnt]=s;head[u]=cnt;
	to[++cnt]=u;net[cnt]=head[v];edge[cnt]=0;cost[cnt]=-s;head[v]=cnt;
}
struct elemt{
	int p,v;
};
struct comp{
	public:
		bool operator()(elemt v1,elemt v2){
			return v1.v<v2.v;
		}
};
-----------------------------------
求[1,MAXN]组合式和逆元 
ll mi(ll a,ll b){
	ll res=1;
	while(b){
		if(b%2){
			res=res*a%mod;
		}	
		a=a*a%mod;
	}
	return res;
}
ll fac[MAXN],inv[MAXN]
fac[0]=1;inv[0]=1;
for(int i=1;i<=MAXN;i){
	fac[i]=(fac[i-1]*i)%mod;
	inv[i]=mi(fac[i],mod-2);
}
ll C(int m,int n){//组合式C(m,n); 
	if(!n){
		return 1;
	}
	return fac[m]*(inv[n]*inv[m*-n]%mod)%mod;
}
---------------------------------
 unordered_map<int,int>mp;
//优先队列默认小顶堆 , greater<int> --小顶堆  less<int> --大顶堆  
priority_queue<elemt,vector<elemt>,comp>q;
	set<int>::iterator it=st.begin();
*/
// vector<vector<int>>edge; 二维虚拟储存坐标 
//-----------------------------------------------------------------------------------------------------------------*/
 // map<int,bool>mp[N]; 
ll dp[N][20*N];//以点i结尾总和为j的方案数
int a[N];
int main(){
/*cout<<setiosflags(ios::fixed)<<setprecision(8)<<ans<<endl;//输出ans(float)格式控制为8位小数(不含整数部分)*/
/*cout<<setprecision(8)<<ans<<endl;//输出ans(float)格式控制为8位小数(含整数部分)*/
	ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);//同步流
int n;
cin>>n;
for(int i=1;i<=n;i++){
	cin>>a[i];
}
int base=10000;
for(int i=1;i<=n;i++){
	dp[i][base+a[i]]++;//从点i出发
	dp[i][base-a[i]]++;
	for(int j=0;j<=20000;j++){//延续上一个点
		if(j-a[i]>=0){
			dp[i][j]=(dp[i][j]+dp[i-1][j-a[i]])%mod;//当前取正
		}
		if(j+a[i]<=20000){
			dp[i][j]=(dp[i][j]+dp[i-1][j+a[i]])%mod;//当前取负
		}
	}
}
ll ans=0;
for(int i=1;i<=n;i++){
	ans=(ans+dp[i][base])%mod;
}
cout<<ans<<endl;

	return 0;
}

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值