简要题意
-
a 为长度为 n( 1 ≤ n ≤ 2000 1≤n≤2000 1≤n≤2000)的序列,满足以下性质
性质1: a i ≤ a i + 1 ( 1 ≤ i ≤ n − 1 ) a_i\le a_{i+1}(1\le i\le n-1) ai≤ai+1(1≤i≤n−1)
性质2: 0 ≤ a [ i ] ≤ 100000 ( 1 ≤ i ≤ n ) 0\le a[i]\le 100000(1\le i\le n) 0≤a[i]≤100000(1≤i≤n)
-
现给出可重集 S(无序)试还原a
-
S=a的所有子段和( 1 ≤ 长 度 L ≤ n 1≤长度L≤n 1≤长度L≤n)
思路
-
首先,看数据范围,初步推断效率大概在 n 2 n^2 n2,看到S无序,想到可能应该将S排序, n 2 ∗ l o g ( n ) ≤ 1 0 8 n^2*log(n)≤10^8 n2∗log(n)≤108。
-
我们可以将s中的元素分为 n n n个组,依据为由几个 a a a中元素共同构成,即 s [ j ] s[j] s[j]的L
-
当 L = 1 L=1 L=1时 s [ j ] s[j] s[j]就是 a [ i ] a[i] a[i],于是 S S S中必定包含 a a a中所有数,又因为性质2,所以在排序后S中的a中元素也是有序出现的,当我们遇见一个数满足所有性质时,它就可能是a中元素。
-
最后一点也是最重要的一点:如何判断一个数不是a中元素呢?
-
其实从数据范围就可以看出来, a [ i ] ≤ 1 0 5 a[i]\le10^5 a[i]≤105,想到可以使用桶标记来排除a中元素与回声重复的情况,但多个a中元素加起来可能会大于 1 0 5 10^5 105,最大可达 2 ∗ 1 0 8 2*10^8 2∗108,那么要不要开一个 2 ∗ 1 0 8 2*10^8 2∗108的桶呢?
-
其实不用,由于性质2,若 s [ i ] ≥ 1 e 5 − 1 s[i]\ge1e5-1 s[i]≥1e5−1,则其必定不是a中元素,也没必要记录了,可以直接排除重复的可能,所以只用一个 1 0 5 10^5 105的桶就可以了。
代码
#include<cstdio>
#include<algorithm>
#define N 2009
#define M 100009
using namespace std;
int n,m,t,k,tot;
int a[M],b[M],ans[N];
int main()
{
scanf("%d",&n);
for(int i=1;i<=(n*(n+1)/2);i++)
{
scanf("%d",&t);
if(t<=100000)
//由于我们只是由前面的a[1~i]推出s[j]是否存在于a中
//所以若s[j]>1e5,便可以判定不在a中,没必要记录。
a[++m]=t;
}
sort(a+1,a+m+1);
//当然如果没有性质2中的非负性,排序也没用。
for(int i=1;i<=m;i++){
if(a[i]>100000)
break;
//由于性质2所以若s[i]>1e5,则其必定不是a中元素,而且经过排序,a[i]≤a[i+1],可以直接退出。
if(b[a[i]]){
b[a[i]]--;
//别忘了消耗桶标记,因为最初的声音与回声可能重复
//如a=1,2,3时,s=1,2,3,3,5,6,这里(1+2)=3的回声和最初的声音3就出现了重复
}
else{
tot=a[i];
ans[++k]=a[i];
for(int j=k-1;j>0;--j){
tot=tot+ans[j];
if(tot<=100000)
b[tot]++;//桶记录
else break;
//由于性质2所以若s[i]>1e5,则其必定不是a中元素
//因为可以通过性质2来判断,所以也没必要记录来浪费空间,可以直接退出。
}
}
}
for(int i=1;i<=k;i++)
printf("%d ",ans[i]);
return 0;
}
其他优化说明
其他优化已经在代码中做了讲解,这里就不列出了
- 由于该题输入比较多 n 2 n^2 n2,所以可以使用快读优化,优化效果比较明显。
namespace IO{
char ibuf[(1<<20)+1],*iS,*iT;
#if ONLINE_JUDGE
#define gh() (iS==iT?iT=(iS=ibuf)+fread(ibuf,1,(1<<20)+1,stdin),(iS==iT?EOF:*iS++):*iS++)
#else
#define gh() getchar()
#endif
#define reg register
inline int read(){
reg char ch=gh();
reg int x=0;
reg char t=0;
while(ch<'0'||ch>'9') t|=ch=='-',ch=gh();
while(ch>='0'&&ch<='9') x=x*10+(ch^48),ch=gh();
return t?-x:x;
}
}
using IO::read;