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

时间:2023-03-08 16:01:02

我自己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;
}