PKU3321

开始写的是用线段树的模型 由于结构的差异 线段较烦 节点中只能记录父节点 然后在维护节点中COUNT值的时候 需要不断地从叶节点开始从下到上对count进行更新 正好和线段树相反 结果超时了
这是因为 如果数据中的树是一棵很高的树的话 维护COUNT值会耗时很大
下面是上面思想的超时代码
(用邻接表也许更省时一点,不过想想如果用精简的树状数组会更美观一些)
#include < stdio.h >
#include
< string .h >

struct  Tree
{
    
int count;
    
bool have;
    
int f;
}
;

Tree fork[
100010 ];
int  con[ 100010 ][ 2 ];
bool  b[ 100010 ];
int  count[ 100010 ];
bool  v[ 100010 ];
int  n, m;
char  op;
int  opN;

void  build( int  f)
{
    fork[f].have
=1;
    fork[f].count
=1;
    
int i;
    
for(i=0;i<n-1;i++)
    
{
        
if(b[i]==0)
        
{
            
if(con[i][0]==f)
            
{
                b[i]
=1;
                fork[con[i][
1]].f=f;
                build(con[i][
1]);
            }

            
if(con[i][1]==f)
            
{
                b[i]
=1;
                fork[con[i][
0]].f=f;
                build(con[i][
0]);
            }

        }

    }

}



void  buildLeaf()
{
    
bool end=1;
    memset(count,
0,sizeof(count));
    
int i;
    
for(i=0;i<n-1;i++)
    
{
        
if(v[con[i][0]]==0 && v[con[i][1]]==0 )
        
{
            count[con[i][
0]]++;
            count[con[i][
1]]++;
            end
=0;
        }

    }

    
for(i=2;i<=n;i++)
    
{
        
if(count[i]==1)
        
{
            fork[fork[i].f].count
+=fork[i].count;
            v[i]
=1;
        }

    }

    
if(!end)
        buildLeaf();
}


void  opera()
{
    
if(op=='C')
    
{
        
if(fork[opN].have==1)
        
{
            fork[opN].have
=0;
            fork[opN].count
--;
            
int f=fork[opN].f;
            
while(f>=1)
            
{
                fork[f].count
--;
                
if(f>1)
                    f
=fork[f].f;
                
else
                    
break;
            }

        }

        
else
        
{
            fork[opN].have
=1;
            fork[opN].count
++;
            
int f=fork[opN].f;
            
while(f>=1)
            
{
                fork[f].count
++;
                
if(f>1)
                    f
=fork[f].f;
                
else
                    
break;
            }

        }

    }

    
else
    
{
        printf(
"%d\n",fork[opN].count);
    }

}



int  main()
{
    
while(scanf("%d",&n)==1)
    
{
        memset(b,
0,sizeof(b));
        
int i;
        
for(i = 0; i < n-1 ; i++)
        
{
            scanf(
"%d%d",&con[i][0],&con[i][1]);
        }

        build(
1);
        memset(v,
0,sizeof(v));
        buildLeaf();
        scanf(
"%d",&m);
        
for(i=0;i<m;i++)
        
{
            scanf(
"\n%c%d"&op, &opN);
            opera();
        }

    }


    
return 0;
}

你可能感兴趣的:(pku)