牛客 小M和天平

題目:https://ac.nowcoder.com/acm/problem/13586
這題很明顯是一個揹包(做多了可能就有感覺了吧),設f[i][j]爲選了前i個物品能否湊出體積爲j的物品,然後用01揹包亂轉移就可以了。但這題要注意的是不能用一維的揹包寫法,必須用二維,因爲這個揹包會同時往前和往後更新狀態(畢竟石子相加和相減都能湊出物品),用一維的話會導致變成完全揹包(不懂可以手動模擬?),然後就可以寫代碼了

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
using namespace std;
int a[105],f[105][10305],n,m;
int main()
{
	while(scanf("%d",&n)!=EOF)
	{
		f[0][0]=1;
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
		for(int j=0;j<=10000;j++)//揹包,若j能被湊出,那麼j+a[i]和j-a[i]也必定能湊出,因爲石子可以放天平兩邊 
		{
			f[i][j]=f[i-1][j];
		}
		for(int j=0;j<=10000;j++)
		{
			f[i][j]|=f[i-1][j+a[i]];
			f[i][j]|=f[i-1][abs(j-a[i])];
			f[i][j]|=f[i-1][j];
		}
	}
	cin>>m;
	for(int i=1;i<=m;i++)
	{
		int k;
		scanf("%d",&k);
		if(k>10000)//這個是數據範圍決定的 
		{
			printf("NO\n");
			continue;
		}
		if(f[n][k])
		{
			printf("YES\n");
		}
		else
		{
			printf("NO\n");
		}
	}
	}
 } 
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章