2014百度之星初赛第一轮解题报告：party

#include<iostream>
#include<stdlib.h>
#include<string.h>
#include<algorithm>
#include<stdio.h>
#include<vector>
#include<queue>

using namespace std;

const int N = 100010;

int val
,ru
,vis
,res
[2];
vector<int>edge
;
queue<int>que;

int main()
{
int T = 1;
int csnum;
scanf("%d",&csnum);
for(int cs=1;cs<=csnum;cs++)
{
int n;
scanf("%d",&n);

for(int i=1;i<=n;i++)
{
edge.clear();
ru=0;
vis=0;
}
while(que.empty()==false)
que.pop();

for(int i=1;i<=n;i++)
{
int a;
scanf("%d",&a);
edge.push_back(a);
edge[a].push_back(i);
ru++;ru[a]++;
}
for(int i=1;i<=n;i++)
{
scanf("%d",&val);
res[1]=val;
res[0]=0;
}

for(int i=1;i<=n;i++)
{
if(ru==1)
que.push(i);
}

while(que.empty()==false)
{
int x=que.front();
que.pop();
vis[x]=1;
for(int i=0;i<edge[x].size();i++)
{
int y=edge[x];
if(vis[y]==0)
{
ru[y]--;
if(ru[y]==1)
que.push(y);

res[y][1]+=res[x][0];
res[y][0]+=max(res[x][1],res[x][0]);
}
}
}

int ans=0;
for(int i=1;i<=n;i++)
{
if(vis==0)
{
int rs[2][2];
rs[0][0]=rs[0][1]=rs[1][0]=res[0];
rs[1][1]=res[1];
int pre=i;
vis=1;

do{
int x=-1;
for(int j=0;j<edge[pre].size();j++)
{
x=edge[pre][j];
if(vis[x]==0)
break;
}
//                                printf("%d-%d ",pre,x);
//                                puts("");
if(vis[x]==1)
break;
vis[x]=1;

int nrs[2][2];
for(int u=0;u<2;u++)
{
nrs[0]=max(rs[0],rs[1])+res[x][0];
nrs[1]=rs[0]+max(res[x][0],res[x][1]);
}
for(int u=0;u<2;u++)
for(int v=0;v<2;v++)
rs[v]=nrs[v];
pre=x;
}while(true);

int a=max(rs[0][0],rs[0][1]);
int b=rs[1][0];
ans+=max(a,b);
//        printf("%d %d\n",i,max(a,b));
}
}
printf("Case #%d:\n%d\n", T++, ans);
}
return 0;
}