2021蓝桥杯初赛砝码称重

题目链接

打比赛的时候作为一个没打过acm的带一萌新,自然是毫无思路的.

现在看到题目第一反应是dfs暴搜,但一看数据量…emmmm,100,o(2ⁿ)的dfs极限状态百分之百超时.

抱着不撞南墙不回头的态度还是用暴搜写了一次

#include <bits/stdc++.h>
using namespace std;

const int N = 100005;

int n;
int a[N];
bool flag[N];
int ans;

void dfs(int idx,int left,int right)
{
    if(idx == n)
    {
        int p = abs(left - right);
        if(!flag[p])
            ans++;
        flag[p] = true;
        return;
    }
    dfs(idx + 1,left,right);
    dfs(idx + 1,left + a[idx],right);
    dfs(idx + 1,left,right + a[idx]);
}

int main()
{
    flag[0] = true;
    cin >> n;
    for(int i = 0 ; i < n ; i++)
        cin >> a[i];
    dfs(0,0,0);
    cout << ans;
    
    return 0;
}

代码很短很好看,也只过了一半
写得不错,下次别写了

然后就考虑dp,问题就在于怎么dp.

对于这道题的状态,就是用几个砝码能称出多少种重量.用一位数组存状态没法进行下一步的推演,但可以用二维数组下标表示某一个重量能否被称出来,状态只有能(1)或不能(0).统计答案的时候把1到规定的最大值N挨个相加就行.总和不超过10⁵,砝码数量最多100,数据范围是数组装得下的.

对于第i个砝码,有三种操作:

1. 压根不用
2. 放在左边
3. 放在右边

所以就在第i-1个砝码能称出的重量的基础上,对这三种情况依次枚举即可.时间复杂度o(n*10⁵).用set也许会更快,但反正不超时还要什么自行车(

![#include <bits/stdc++.h>
using namespace std;

const int N = 100005;

int n;
int a[N];
int ans[105][N];

int main()
{
    cin >> n;
    for(int i = 0 ; i < n ; i++)
    {
        cin >> a[i];
        if(!i)
        {
            ans[i][a[i]] = 1;
            ans[i][0] = 1;
        }
        else
        {
            for(int j = 0 ; j < N ; j++)
                if(ans[i - 1][j])
                {
                    ans[i][j] = 1;
                    ans[i][j + a[i]] = 1;
                    ans[i][abs(j - a[i])] = 1;
                }
        }
    }
    
    int sum = 0;
    for(int i = 1 ; i < N ; i++)
        sum += ans[n - 1][i];
    cout << sum;
    
    return 0;
}

依然很短很好看,但AC了(