题意:
有n个连续方块和m个颜色操作(l[1],l[2]…l[m]),每次操作允许你连续涂色l[i]个方块,最后必须保证所有颜色都出现在方块上且n个方块都被着色。
想法:
从输入时开始判断可否满足题意。从i=0时开始输入数据,计算l[i]的和sum,且此时必须有(n-l[i]+1)>i。因为如果不满足这个条件意味着,不管你怎么涂,这一次涂完后,剩下的格子不足以装下前面所有的颜色,即前面的颜色至少有一种被当次颜色完全覆盖,不满足所有颜色都出现在方块上。
同时sum>=n也必须满足,不然即使全部操作涂完也填不满方块。
下面开始涂方块:
1.从第一块开始涂,涂完sum-l[i];
2.当sum大于剩余的未着色方块总数时,每次涂就只给前一个颜色留一个颜色块,涂完sum-l[i];
3.当sum等于剩余的未着色方块总数时,每次涂就跟着上一个颜色涂完的方块的后面跟着涂就可以了,涂完sum-l[i];
4.最后一次,从n-l[i]+1开始涂,这样不管怎么涂,都可以使方块被涂满。
代码如下:
#include <bits/stdc++.h>
using namespace std;
const int r=1e5+2;
int n,m;
int l[r];
int main ()
{
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
cin>>n>>m;
long long tot=0;
bool flag=false;
for (int i=0;i<m;i++)
{
cin>>l[i];
tot+=l[i];//记录l[i]的和
if ((n-l[i]+1)<=i)
flag=true;
}
if (tot<n || flag)
cout<<"-1"<<endl;
else
{
int place=1;//记录涂的方块的起始位置
int place_max;//记录每次涂完之后的方块的结束位置
for (int i=0;i<m;i++)
{
cout<<place<<" ";
place_max=place+l[i]-1;
tot-=l[i];
if (i==(m-2))
{
place=n-l[i+1]+1;//最后一次涂的位置
continue;
}
if (tot==(n-place_max))
place=place_max+1;
else
place++;
}
}
return 0;
}