poj 3674 枚举+贪心与搜索

http://acm.pku.edu.cn/JudgeOnline/problem?id=3674

看了别人的代码，好久才领悟到什么意思，于是就写了个 枚举+贪心 但耗时比较多，1400多ms。主要是应在每种级别中选取攻击力前六的，这样得到的规模最大为66.

#include <iostream>

#include <vector>

#include <algorithm>

using namespace std;

vector<int> v[11];

int num[11],b[11],c[7],m;

double res;

struct Node

{

int index,e;

}node[7];

int cmp(int x,int y)

{

return x>y;

}

int cmpp(Node x,Node y)

{

return x.e>y.e;

}

void dfs(int dep)

{

int i;

if(dep==m)

{

node[0].index=0;

node[0].e=-1;

for(i=1;i<m;i++)

{

node[i].index=i;

node[i].e=c[i-1];

}

sort(node,node+m,cmpp);

int dns[11]={0};

int d[7];

for(i=0;i<m;i++)

{

int p=node[i].index;

d[p]=v[c[p]][dns[c[p]]++];

}

double sum=d[0];

for(i=1;i<m;i++)

sum+=d[i]*(1.0+c[i-1]*1.0/10);

if(sum>res)

res=sum;

return;

}

for(i=0;i<11;i++)

{

if(b[i]<num[i])

{

b[i]++;

c[dep]=i;

dfs(dep+1);

b[i]--;

}

}

}

int main()

{

int n,i,d,e;

while(scanf("%d",&n)!=EOF)

{

for(i=0;i<11;i++)

v[i].clear();

m=n>=6?6:n;

while(n--)

{

scanf("%d%d",&d,&e);

v[e].push_back(d);

}

for(i=0;i<11;i++){

sort(v[i].begin(),v[i].end(),cmp);

num[i]=v[i].size();

}

res=0;

memset(b,0,sizeof(b));

dfs(0);

printf("%.0lf\n",res);

}

return 0;

}

看了超哥的提示，突然发现此规模搜索确实比较好，稍微加点剪枝

#include <iostream>

#include <vector>

#include <algorithm>

using namespace std;

vector<int> v[11];

int num[11],c[7],m,l;

bool visit[70];

double res;

struct Node

{

int s,e;

}node[67];

int cmp(int x,int y)

{

return x>y;

}

int cmpp(Node x,Node y)

{

if(x.e==y.e)

return x.s>y.s;

return x.e>y.e;

}

void dfs(int i,double value,int s)

{

if(i==m)

{

if(value>res)

res=value;

return;

}

if(value+(m-i)*node[0].e*2<=res)

return;

int j;

for(j=0;j<l;j++)

{

if(!visit[j])

{

visit[j]=1;

dfs(i+1,value+(1.0+s*1.0/10)*node[j].e,node[j].s);

visit[j]=0;

}

}

}

int main()

{

int n,i,j,d,e;

while(scanf("%d",&n)!=EOF)

{

for(i=0;i<11;i++)

v[i].clear();

m=n>=6?6:n;

while(n--)

{

scanf("%d%d",&d,&e);

v[e].push_back(d);

}

for(i=0;i<11;i++){

sort(v[i].begin(),v[i].end(),cmp);

}

l=0;

for(i=0;i<11;i++)

{

for(j=0;j<v[i].size()&&j<6;j++)

{

node[l].e=v[i][j];

node[l++].s=i;

}

}

sort(node,node+l,cmpp);

res=0;

memset(visit,0,sizeof(visit));

dfs(0,0,0);

printf("%.0lf\n",res);

}

return 0;

}