2019-ICPC-Shanghai-online-J-StoneGame

題目鏈接 https://nanti.jisuanke.com/t/41420

題目大意

給你一些石頭,讓你選一些石頭重量大於剩餘的重量,且去掉任意一個你選的石頭之後總重量小於等於剩餘重量

思路

將石頭按從大到小排序,然後遍歷選取,以當前的石頭爲最小值,也就是當前的石頭一定要選,比它大的可選可不選,比它小的一定不選
然後就是揹包問題了,dp[j]代表選擇石頭的總重量爲j的方案數,判斷一下j合不合法,合法就加上dp[j-num[i]];

AC代碼

#include <iostream>
#include <cstring> 
#include <algorithm> 
#include <cstdio>
using namespace std;
typedef long long ll;
#define wfor(i,j,k) for(i=j;i<k;++i)
#define mfor(i,j,k) for(i=j;i>=k;--i)
// void read(int &x) {
// 	char ch = getchar(); x = 0;
// 	for (; ch < '0' || ch > '9'; ch = getchar());
// 	for (; ch >= '0' && ch <= '9'; ch = getchar()) x = x * 10 + ch - '0';
// }
const int maxn=305;
ll dp[150005];
int num[maxn];
const int mod=1e9+7;
int main()
{
    std::ios::sync_with_stdio(false);
    #ifdef test
    freopen("F:\\Desktop\\question\\in.txt","r",stdin);
    #endif
    #ifdef ubuntu
    freopen("/home/time/debug/debug/in","r",stdin);
    freopen("/home/time/debug/debug/out","w",stdout);
    #endif
    int t;
    cin>>t;
    while(t--)
    {
        memset(dp,0,sizeof(dp));
        int n;
        cin>>n;
        int i;
        int sum=0;
        wfor(i,1,n+1)
        {
            cin>>num[i];
            sum+=num[i];
        }
        sort(num+1,num+n+1);
        int j;
        ll ans=0;
        dp[0]=1;
        mfor(i,n,1)
        {
            mfor(j,sum,num[i])
            {
                int rest=sum-j;
                if(j>=rest&&j-num[i]<=rest)
                {
                    ans+=dp[j-num[i]];
                    ans%=mod;
                }
                dp[j]+=dp[j-num[i]];
                dp[j]%=mod;
            }
        }
        cout<<ans<<endl;
    }
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章