【NOIP2003】传染病控制(-贪心/dfs)

我自己yy了个贪心算法,在某oj 0msAC~。然后去wikioi提交,呵呵,原来是之前oj的数据太弱给我水过了,我晕。

我之前的想法是在这棵树上维护sum,然后按时间来割边,每一时刻割已经感染的人所连接的sum值最大的边。,,才60分。。

#include <cstdio>

#include <cstring>

#include <cmath>

#include <string>

#include <iostream>

#include <algorithm>

using namespace std;

#define rep(i, n) for(int i=0; i<(n); ++i)

#define for1(i,a,n) for(int i=(a);i<=(n);++i)

#define for2(i,a,n) for(int i=(a);i<(n);++i)

#define for3(i,a,n) for(int i=(a);i>=(n);--i)

#define for4(i,a,n) for(int i=(a);i>(n);--i)

#define CC(i,a) memset(i,a,sizeof(i))

#define max(a,b) ((a)>(b)?(a):(b))

#define min(a,b) ((a)<(b)?(a):(b))

#define read(a) a=getnum()

#define print(a) printf("%d", a)

#define debug(x) printf("debug: %d\n", x)

#define printarr(a, x) { for1(i, 0, x) printf("%d ", a[i]); printf("\n"); }

inline int getnum() { int ret=0; char c; int k=1; for(c=getchar(); c<'0' || c>'9'; c=getchar()) if(c=='-') k=-1; for(; c>='0' && c<='9'; c=getchar()) ret=ret*10+c-'0'; return ret*k; }



const int N=305;

int ihead[N], inext[N*N], to[N*N], cnt, sum[N], n, m, fa[N];

bool vis[N], die[N], vis2[N];



inline void pushup(int x) { sum[x]=1; for(int i=ihead[x]; i; i=inext[i]) if(to[i]!=fa[x]) sum[x]+=sum[to[i]]; }

inline void add(int u, int v) {

	inext[++cnt]=ihead[u]; ihead[u]=cnt; to[cnt]=v;

	inext[++cnt]=ihead[v]; ihead[v]=cnt; to[cnt]=u;

}



void build(const int &x) {

	if(vis[x]) return; vis[x]=true;

	for(int i=ihead[x]; i; i=inext[i]) if(!vis[to[i]]) fa[to[i]]=x, build(to[i]);

	pushup(x);

}



int main() {

	bool flag;

	int maxi, v, ans=0;

	read(n); read(m);

	rep(i, m) add(getnum(), getnum());

	build(1);

	CC(vis, 0);

	vis[1]=die[1]=true;

	int fff=sum[1];

	for1(ttt, 1, fff) {

		flag=1; maxi=0;

		memcpy(vis2, vis, sizeof(vis));

		for1(u, 1, n) if(vis2[u]) {

			for(int i=ihead[u]; i; i=inext[i]) if(to[i]!=fa[u] && !vis[v=to[i]]) {

				die[v]=1;

				flag=0;

				if(sum[v]>sum[maxi]) maxi=v;

				vis[v]=true;

			}

			vis[u]=vis[maxi]=die[maxi]=false;

		}

		if(flag) break;

	}

	for1(i, 1, n) if(die[i]) ans++;

	print(ans);

	return 0;

}

 

 

好吧,看题解。。

恩,,,深搜,因为贪心那个建图的话,不能处理环 囧。

(或许某天yy出处理环的贪心0.0)

(不行,我得思考一下我的贪心,好像似乎可行。

先放出我后边写的dfs。。

#include <cstdio>

#include <cstring>

#include <cmath>

#include <string>

#include <iostream>

#include <algorithm>

using namespace std;

#define rep(i, n) for(int i=0; i<(n); ++i)

#define for1(i,a,n) for(int i=(a);i<=(n);++i)

#define for2(i,a,n) for(int i=(a);i<(n);++i)

#define for3(i,a,n) for(int i=(a);i>=(n);--i)

#define for4(i,a,n) for(int i=(a);i>(n);--i)

#define CC(i,a) memset(i,a,sizeof(i))

#define max(a,b) ((a)>(b)?(a):(b))

#define min(a,b) ((a)<(b)?(a):(b))

#define read(a) a=getnum()

#define print(a) printf("%d", a)

#define debug(x) printf("debug: %d\n", x)

#define printarr(a, x) { for1(i, 0, x) printf("%d ", a[i]); printf("\n"); }

inline int getnum() { int ret=0; char c; int k=1; for(c=getchar(); c<'0' || c>'9'; c=getchar()) if(c=='-') k=-1; for(; c>='0' && c<='9'; c=getchar()) ret=ret*10+c-'0'; return ret*k; }



const int N=305;

int ihead[N], inext[N*N], to[N*N], fa[N], d[N], cnt, n, m, ans=~0u>>1;

bool die[N], vis[N];



inline void add(int u, int v) {

	inext[++cnt]=ihead[u]; ihead[u]=cnt; to[cnt]=v;

	inext[++cnt]=ihead[v]; ihead[v]=cnt; to[cnt]=u;

}



void build(const int &x) {

	if(vis[x]) return; vis[x]=true;

	for(int i=ihead[x]; i; i=inext[i]) if(!vis[to[i]]) {

		fa[to[i]]=x; d[to[i]]=d[x]+1;

		build(to[i]);

	}

}



void dfs(const int &dis, int num) {

	if(num>=ans) return;

	bool flag=1;

	for1(u, 1, n) if(d[u]==dis && die[u])

		for(int i=ihead[u]; i; i=inext[i])

			if(fa[to[i]]==u) flag=0, die[to[i]]=1, ++num;

	--num;

	for1(u, 1, n) if(d[u]==dis+1 && die[u]) die[u]=0, dfs(dis+1, num), die[u]=1;

	++num;

	for1(u, 1, n) if(d[u]==dis && die[u])

		for(int i=ihead[u]; i; i=inext[i])

			if(fa[to[i]]==u) die[to[i]]=0, --num;

	if(flag && ans>num) ans=num;

}



int main() {

	read(n); read(m);

	rep(i, m) add(getnum(), getnum());

	die[1]=true; d[1]=1;

	build(1);

	dfs(1, 1);

	print(ans);

	return 0;

}

 

你可能感兴趣的:(2003)