Educational Codeforces Round 145 (Rated for Div. 2)C. Sum on Subarrays(构造)

很意思的一道构造题
Educational Codeforces Round 145 (Rated for Div. 2)C. Sum on Subarrays(构造)_第1张图片

题意:给一个 n 、 k n、k nk,让构造长度为n的数组满足,子数组为整数的个数为k个,负数的为 k − ( n + 1 ) ∗ n / 2 k-(n+1)* n/2 k(n+1)n/2,每个数的范围为 [ − 1000 , 1000 ] [-1000,1000] [1000,1000]

这种构造题可以考虑就是前一段可以一直用一样的、最小的。
我们观察可以发现 k + k − ( n + 1 ) ∗ n / 2 = ( n + 1 ) ∗ n / 2 k+k-(n+1)* n/2= (n+1)* n/2 k+k(n+1)n/2=(n+1)n/2
也就是所有子数组的个数,换句话说子数组不能有0。
这样我们很容易考虑用很小的一个负数和一个很小的正数去构造
这里我用的是 1 1 1 − 1000 -1000 1000
我们先考考虑一下前一段是p个1,后面全是-1000的情况这样我们得到的正数组有 ( p + 1 ) ∗ p 2 个 \frac{(p+1) * p}{2}个 2(p+1)p
k = ( p + 1 ) ∗ p 2 k=\frac{(p+1) * p}{2} k=2(p+1)p时,自然皆大欢喜
k > = ( p + 1 ) ∗ p 2 k>=\frac{(p+1) * p}{2} k>=2(p+1)p时,我们考虑一下剩下的 k − ( p + 1 ) ∗ p 2 k-\frac{(p+1) * p}{2} k2(p+1)p该如何臭凑出来,能增加p吗?,当p+1,我们会增加p+1个正数组,这是不行的,我们考虑的p的最大满足 k > = ( p + 1 ) ∗ p 2 k>=\frac{(p+1) * p}{2} k>=2(p+1)p的p,也就是说缺少的正数组个数是在 [ 1 , p ] [1,p] [1,p]
我们可以选择前面p个1中的一个将其变为1000, p + 1 p+1 p+1处的-1000遍为500,这样我们就可以添加 [ 1 , p ] [1,p] [1,p]个正数组,哪个位置的1变为1000呢?
我们可以找一下规律
Educational Codeforces Round 145 (Rated for Div. 2)C. Sum on Subarrays(构造)_第2张图片
弄清楚上面的事情,代码就很简单了,我们只需要而分出最后一个满足条件的p然后按照上面的构造方法放数即可

#include  
#define int long long
#define rep(i,a,b) for(int i = (a); i <= (b); ++i)
#define fep(i,a,b) for(int i = (a); i >= (b); --i)
#define pii pair<int, int>
#define pll pair<long long, long long>
#define ll long long
#define db double
#define endl '\n'
#define x first
#define y second
#define pb push_back

using namespace std;

const int N=5e3+10,mod=100003,inf=(1ull<<63)-1;
int n,m,k;
int vis[N],d[N];
int a[1010],b[1010];



void solve()
{
	cin>>n>>k;
	int l=0,r=n;
	while(l<r){
		int mid=(l+r+1)>>1;
		if(mid*(mid+1)/2<=k)	l=mid;
		else	r=mid-1;
	}
	if(l*(l+1)/2==k){
		rep(i,1,l)	cout<<1<<' ';
		rep(i,l+1,n){
			if(i==l+1)	cout<<-500<<' ';
			else cout<<-1000<<' ';
		}
		cout<<endl;	
	}else{
		int d=k-(l*(l+1))/2;
		rep(i,1,l){
			if(i==d)	cout<<1000<<' ';
			else	cout<<1<<' ';
		}
		rep(i,l+1,n){
				if(i==l+1)	cout<<-500<<' ';
				else cout<<-1000<<' ';
			}	
		cout<<endl;	
	}
}

signed main(){
	ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
//   	freopen("1.in", "r", stdin);
  	int _;
	cin>>_;
	while(_--)
	solve();
	return 0;
}

你可能感兴趣的:(codeforces,c语言,spring,boot,django,fastapi,maven,spring,java)