TopCoder SRM 583 TurnOnLamps

时间:2022-05-12 14:48:32

读错题了有没有呀,原来 lamps 是在边上的呀,当成是在点上的了,无语。

直接一个dfs 就可以 从叶子节点开始,如果有必要转换 lamp 的状态则加一个仅包含 这个 lamp 的段

然后向上扩展,对于某个子树,根据扩展上来的段的个数,将偶数段进行合并(这是最优选择),然后可能剩

一个或者0个段没用上,再根据这个节点到父节点的lamp的状态进行分析是否要将剩下的段向上扩展。

直到根节点。

由于所给边的特殊性,可以直接遍历。

代码:

#include<iostream>
#include<cstdio>
#include<vector>
#include<stack>
#include<cstring>
#include<queue>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<string>
#include<set>
#include<map> #define ll long long
#define ull unsigned long long
using namespace std;
const int INF=0x3f3f3f3f;
/*
class TurnOnLamps
{
public:
int minimize(vector <int> r, string S, string I)
{
int x[100]={0};
int n=r.size();
int sum=0;
for(int i=n-1;i>=0;--i)
{
if(I[i]=='1')
{
if((S[i]=='0'&&x[i+1]==0)||(S[i]=='1'&&x[i+1]==1))
{
++sum;
x[i+1]^=1;
}
}
x[r[i]]^=x[i+1];
}
return (sum+1)/2;
}
};
*/
const int N=100;
int head[N],cnt;
struct node
{
int j,next;
int state,im;
}edge[N];
int dp[N];
int x[N];
void add(int i,int j,int state,int im)
{
edge[cnt].j=j;
edge[cnt].state=state;
edge[cnt].im=im;
head[i]=cnt++;
}
int dfs(int x,int S,int I)
{
int sum=0;
for(int t=head[x];t!=-1;t=edge[t].next)
{
sum+=dfs(edge[t].j,edge[t].state,edge[t].im);
}
dp[x]=sum/2;
int tmp=sum%2;
if(I==0)
return tmp;
else
{
if(S==1&&tmp==1)
{dp[x]+=1;return 0;}
if(S==1&&tmp==0)
return 0;
if(S==0&&tmp==0)
return 1;
if(S==0&&tmp==1)
return 1;
}
return 0;
}
class TurnOnLamps
{
public:
int minimize(vector <int> r, string S, string I)
{
memset(head,-1,sizeof(head));cnt=0;
for(unsigned int i=0;i<r.size();++i)
{
add(r[i],i+1,S[i]-48,I[i]-48);
}
int tmp=dfs(0,0,0);
for(unsigned int i=0;i<r.size()+1;++i)
tmp+=dp[i];
return tmp;
}
};