AC自动机

[蓝桥杯 2021 省 AB] 砝码称重

题目描述:

你有一架天平和 N 个砝码, 这 N 个砝码重量依次是 W1, W2 .. ..WN 。 请你计算一共可以称出多少种不同的重量?

注意砝码可以放在天平两边。

思路:总共可选择的重量在1到总重之间。f[i][j] 表示在前i个砝码中重量为j.

可分为四种状态:

1.当 j==a[i] f[i][j]=1。

2.当f[i-1][j]==1 f[i][j]=1. 无需添加。

3.当 f[i-1][j+a[i]]==1 f[i][j]=1.可添加一个当前砝码。

4.当 f[i-1][j-a[i]]==1 f[i][j]=1.可减少一个当前砝码。

复制代码
#include <iostream>
using namespace std;
int a[110];
int f[110][100010];
int main()
{
	int n;
	cin>>n;
	int sum=0;
	for (int i=1;i<=n;i++){
		cin>>a[i];
		
		sum+=a[i];
	}
	
	for (int i=1;i<=n;i++){
		for (int j=sum;j;j--){
			if(j==a[i])f[i][j]=1;
			else if (f[i-1][j]==1)f[i][j]=1;
			else if (f[i-1][j+a[i]]==1)f[i][j]=1;
			else if (f[i-1][abs(j-a[i])]==1) f[i][j]=1;
		}
	}
	int ans=0;
    
	for (int i=1;i<=sum;i++){
		if(f[n][i]==1){
			ans++;
		}
	}

	cout<<ans;
	
	
}
相关推荐
努力的小帅2 个月前
C++_STL_map与set
开发语言·数据结构·c++·学习·leetcode·刷题
Belgnas2 个月前
刷题:(题目名称)(模板)
刷题
愚润求学2 个月前
【专题四】前缀和(3)
开发语言·c++·笔记·leetcode·刷题·c++11
愚润求学2 个月前
【专题刷题】二分查找(一):深度解刨二分思想和二分模板
开发语言·c++·笔记·leetcode·刷题
Belgnas2 个月前
刷题:奇怪的电梯
刷题
愚润求学2 个月前
【专题刷题】滑动窗口(四):
c++·笔记·leetcode·刷题
愚润求学3 个月前
【专题刷题】双指针(三):两数之和,三数之和,四数之和
c++·笔记·leetcode·刷题
修修修也3 个月前
【算法手记7】拼三角
数据结构·学习·算法·刷题
veejaLiu4 个月前
LeetCode 30.串联所有单词的子串
java·学习·算法·leetcode·职场和发展·刷题
修修修也4 个月前
算法手记3
数据结构·学习·算法·刷题