题目大意:一个括号字符数组S,可以用两个整数数组P和W来描述,其中数组P中的元素Pi定义为:第i个右括号前面有Pi个左括号;数组W中的元素Wi定义为,第i个右括号和其配对的左括号之间所包含的完整的括号的对数。现在给你P数组,让你求出该字符数组S所对应的W数组。
分析:我们可以由P数组先求出S数组,再由S数组找出W数组会更容易些。至于怎么由P数组找出S呢,我们用0和1分别表示左括号和右括号,先把S数组初始化为0(即初始化为只包含左括号),那么对于每一个输入的Pi,由定义我们知道,这是第i个右括号,同时我们还知道,这个右括号前面有Pi个左括号,这时我们就可以对S[ Pi+i ]赋值为1,即该点为右括号。有了S数组,找W数组就很简单了。
实现代码如下:
#include <cstdio>
#include <iostream>
#include <cstring>
using namespace std;
int main()
{
int t,n,i,j,k;
int p,w[25],s[50];
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
memset(s,0,sizeof(s));
for(i=0;i<n;i++)
{
scanf("%d",&p);
s[p+i]=1;
}
k=0;
for(i=0;i<50;i++)
if(s[i]==1)
{
int l=0,r=1;
for(j=i-1;j>=0;j--)
{
if(s[j]==0) l++;
if(s[j]==1) r++;
if(s[j]==0&&l==r)
{
w[k++]=l;
break;
}
}
}
for(i=0;i<n;i++)
printf("%d ",w[i]);
printf("\n");
}
return 0;
}