Ozon Tech Challenge 2020 E. Kuroni and the Score Distribution(思维)

题目链接：https://codeforces.com/contest/1305/problem/E

 

题目大意：

  构造n个数，满足存在m组$a[i]+a[j]=a[k],i\le j\le k,1\le a[i]\le 1e9,n\le 5000$

 

题目思路：

  首先自己写一下可以发现，对于1~n的情况下，n取1,2,3,4,5…形成的组数分别为：0,0,1,2,4,6,9,12,16,20,25…然后就可以发现，每加入一个数字，相较于前一个数字的结果会多加入$(n-1)/2$组情况。如果对于位于两者之间，就是再加一个数字就超了，不加又不够的情况，这个时候就对于剩余需要的组数m，希望再加入一个数字，这个数字的贡献正好为m，把它补全。那就是a[i-1]+a[i-2*m]，正好有m组可以得到加入的数字。如果组数够了，n不够咋办呢，加入前一个数+5000，因为一共也就5000个数，不可能到5000，如果新加的数字都差5000，就不会形成新的组。

 

以下是代码：

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

#define ll long long
#define rep(i,a,b) for(ll i=a;i<=b;i++)
#define per(i,a,b) for(ll i=a;i>=b;i--)
const ll MAXN =2e5+5;
const ll MAXM = 4e7+5;
const ll MOD = 998244353;
int n,m;
int ans[MAXN];
int main()
{
    while(~scanf("%d%d",&n,&m)){
        ans[1]=1,ans[2]=2;
        rep(i,3,n){
            if(!m){
                ans[i]=ans[i-1]+5000;
                continue;
            }
            if(m>=(i-1)/2){
                m-=(i-1)/2;
                ans[i]=ans[i-1]+1;
            }
            else{
                ans[i]=ans[i-1]+ans[i-2*m];
                m=0;
            }
        }
        if(m)puts("-1");
        else{
            rep(i,1,n){
                printf("%d%c",ans[i],i==n?'\n':' ');
            }
        }
    }
}