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

很意思的一道构造题
image

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

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

#include <bits/stdc++.h> 
#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;
}

posted @ 2024-02-09 12:12  cxy8  阅读(7)  评论(0编辑  收藏  举报