本题我们用C++中的STL容器来做,通过vector和set(去重功能)来存放砝码,先初始化vector容器后再进行遍历,最后输出即可。
#include<bits/stdc++.h>
using namespace std;
int main()
{
int n;
cin>>n;
vector<int>a(n);
for(int i=0;i<n;i++)
{
cin>>a[i];
}
set<int>book;
book.insert(0); //初始化set集合
set<int>::iterator it;
for(int i=0;i<n;i++)
{
vector<int>b; //构建临时容器来存储
for(it=book.begin();it!=book.end();it++)
{
int t1=*it+a[i]; //砝码的右边放数
int t2=*it-a[i]; //砝码的左边放数
b.push_back(t1);
b.push_back(t2);
}
for(int j=0;j<b.size();j++)
{
book.insert(fabs(b[j])); //注意砝码不能为负数,加到set中去重
}
}
cout<<book.size()-1<<endl; //最后把开始加上的0元素剪掉
return 0;
}
注意在调试程序的过程中,在枚举完每一种情况后,更新book数组中的内容,每次更新完后再重新往砝码两端再次放入砝码,vector b就是这个更新的作用。(长时间没用STL了,还得去温习一下STL的用法,不过用STL是真的香啊~)。时间复杂度为O(n^2),数据规模是1e5,所以总的就是1e10,是直接可以过的。