题目大意
有n层蛋糕每个第i层上要涂上一层单位面积为ai的奶油,奶油会顺着蛋糕侧边流下,每ai面积的奶油就可以覆盖ai层蛋糕,你最后需要输出哪些层有奶油覆盖哪些层没有。
输入格式
第一行输入t(1<=t<=20000),表示t组测试样例。
接下来每次测试样例的第一行输入一个n(1<=n<=2e5),表示有n层蛋糕。
然后输入n个数,表示从第1层到第n层每层上面覆盖的奶油的面积。
输出格式
依次从第1行到第n行输出1或0,1表示该层被奶油覆盖,0表示该层没有被奶油覆盖。
样例
input:
3
6
0 3 0 0 1 3
10
0 0 0 1 0 5 0 0 0 2
3
0 0 0
output:
1 1 0 1 1 1
0 1 1 1 1 1 0 0 1 1
0 0 0
思路
这道题就是一道很常规的dp,但是我居然写的时候没看出来。。。果然还是太菜了吗,解法就是运用dp的思想,从n层开始,记录奶油可以流到的最下面的位置,用一个变量mn来记录,mn=min(mn,i-a[i]),然后每层都用dp数组记录一下此时的mn,最后遍历一遍dp数组,如果dp[i]<=i,就说明该层可以被覆盖。
代码
//dp
#include<cstdio>
using namespace std;
const int MAXN = 200005;
int arr[MAXN];
int dp[MAXN];
int myMin(int a,int b)
{
if(a<=b) return a;
else return b;
}
int main()
{
int t;
while(scanf("%d",&t)!=EOF)
{
while(t--)
{
int n;
scanf("%d",&n);
for(int i = 0;i < n;i++)
scanf("%d",&arr[i]);
int mn = n;
for(int i = n-1;i >= 0;i--)
{
mn = myMin(mn,i-arr[i]+1);
dp[i] = mn;
}
for(int i = 0;i < n;i++)
{
if(dp[i]<=i)
printf("1 ");
else
printf("0 ");
}
printf("\n");
}
}
return 0;
}