题意:
p数组的意义:记录第i个右括号前有几个左括号 (注意是之前一共有多少左括号)
w数组的意义:记录第i个右括号中包含了多少个括号(包括他本身)
根据p数组求w数组(注意input中的n代表括号对数,不是半个括号的数量)
例子说明w数组:
(()())
第一个右括号只包含自己,故为 1;
第二个右括号也只包含自己, 故为 1;
第三个右括号包含前面两个括号,再加上自己,故为 3.
第二个右括号也只包含自己, 故为 1;
第三个右括号包含前面两个括号,再加上自己,故为 3.
算法:
用1代表左括号,用2代表右括号,先通过p数组初始化s数组。用numOfTwo记录右括号在s(也就是2)的位置,对每一个右括号遍历:用counts记录右括号的个数,往前遍历,遇到左括号counts--(代表抵消一个括号),遇到右括号counts++(代表多了一个需要抵消的括号),直到遍历完或者counts==0(代表所有右括号都被左括号抵消了)
AC代码:
#include <iostream>
#include <cstring>
using namespace std;
int main()
{
int cases;
cin>>cases;
while(cases--)
{
int num;
cin>>num;
int s[45];
memset(s,0,sizeof(s));
int p[25];
for(int i=0;i<num;i++)
{
cin>>p[i];
}
int w[25];
int numOfTwo[25];
for(int i=0;i<num;i++) //先初始化s中的2的位置
{
s[i+p[i]]=2;
numOfTwo[i]=i+p[i]; //记录右括号在s中的位置
}
for(int i=0;i<41;i++) //再初始化剩余的左括号
{
if(s[i]!=2)
s[i]=1;
}
for(int i=0;i<num;i++) //遍历s中的每个右括号
{
int counts=0,ans=0;
int a=numOfTwo[i];
counts++;
for(int j=a-1;j>=0;j--) //在每个右括号基础上往前遍历
{
if(s[j]==1) //右括号被左括号抵消
{
ans++;
counts--;
}
else if(s[j]==2) counts++; //遇到新的右括号
if(counts==0) //右括号全部被抵消
{
w[i]=ans; //记录ans
break;
}
}
}
int i;
for(i=0;i<num-1;i++) //输出答案
cout<<w[i]<<" ";
cout<<w[i]<<endl;
}
return 0;
}