http://acm.hdu.edu.cn/showproblem.php?pid=6140
有n个物体,每个物体有一个权值,问能否选择部分物体使得组成的权值为k
每个物体有不同类型,L:权值为正,D:权值为负,N:权值可正可负
并且a1=1,类型为N,ai<sum(类型为N和和自己同类型的物体权值aj) 且j<i
因为a1=1,类型为N,通过a1可以得到-1,0,1的状态,而有了“ai<sum(类型为N和和自己同类型的物体权值aj) 且j<i”这个条件,每次都可以得到连续的可行的状态
最后判一下就可以了
#include<bits/stdc++.h>
using namespace std;
int a[1005];
char b[1005];
int main()
{
int T,n,k,i,j;
cin>>T;
while(T--)
{
cin>>n>>k;
for(i=0;i<n;i++)
{
scanf("%d",&a[i]);
}
for(i=0;i<n;i++)
{
scanf(" %c",&b[i]);
}
int le=-1,ri=1;
for(i=1;i<n;i++)
{
if(b[i]!='D')
{
ri+=a[i];
}
if(b[i]!='L')
{
le-=a[i];
}
}
if(k<=ri&&k>=le)
cout<<"yes"<<endl;
else
cout<<"no"<<endl;
}
return 0;
}
想复杂了,写背包卡去了。。。。。。
利用一个后缀优化一下背包的范围。。。。500+ms卡过
和正解差距太大
#include<bits/stdc++.h>
using namespace std;
int dp[2000505];
int a[1005];
char b[1005];
int minn[1005],maxx[1005];
int pian=1000010;
int main()
{
int T,n,k,i,j;
cin>>T;
while(T--)
{
cin>>n>>k;
for(i=0;i<n;i++)
{
scanf("%d",&a[i]);
}
for(i=0;i<n;i++)
{
scanf(" %c",&b[i]);
}
if(k<0)
{
for(i=0;i<n;i++)
{
if(b[i]=='D')
b[i]='L';
else if(b[i]=='L')
b[i]='D';
}
k*=-1;
}
for(i=n-1;i>=0;i--)
{
minn[i]=minn[i+1];
maxx[i]=maxx[i+1];
if(b[i]=='N'||b[i]=='L')
minn[i]+=a[i];
else if(b[i]=='N'||b[i]=='D')
maxx[i]+=a[i];
}
memset(dp,0,sizeof(dp));
int sum1=pian,sum2=pian;
dp[pian]=1;
for(i=0;i<n;i++)
{
if(b[i]=='N'||b[i]=='L')
{
int down=sum1+a[i],up=sum2+a[i];
for(j=up;j>=down;j--)
{
dp[j]|=dp[j-a[i]];
}
sum2+=a[i];
if(sum2>=k+pian+maxx[i])
sum2=k+pian+maxx[i];
}
if(b[i]=='N'||b[i]=='D')
{
int down=sum1-a[i],up=sum2-a[i];
for(j=down;j<=up;j++)
{
dp[j]|=dp[j+a[i]];
}
sum1-=a[i];
if(sum1<k-minn[i]+pian)
sum1=k-minn[i]+pian;
}
}
if(dp[k+pian])
cout<<"yes"<<endl;
else
cout<<"no"<<endl;
}
return 0;
}