首页 > 代码库 > Poj1015
Poj1015
还以为多牛逼呢,。。就是长了点。。dp弄得像枚举一样。贴一个优秀的代码,
#include<iostream>#include<cmath>#include<cstring>#include<algorithm>using namespace std;int n; //候选人数int m; //当选人数int dp[21][801]; //dp[j][k]:取j个候选人,使其辩控差为k的所有方案中,辩控和最大的方案的辩控和int path[21][801]; //记录所选定的候选人的编号/*回溯,确认dp[j][k]方案是否曾选择过候选人i*/bool select(int j,int k,int i,int* v){ while(j>0 && path[j][k]!=i) { k-=v[ path[j][k] ]; j--; } return j?false:true;}int main(void){ int time=1; while(cin>>n>>m && n) { /*Initial*/ int j,k,i; int* p=new int[n+1]; //每个人的控方值 int* d=new int[n+1]; //每个人的辩方值 int* s=new int[n+1]; //每个人的辨控和 int* v=new int[n+1]; //每个人的辨控差 memset(dp,-1,sizeof(dp)); memset(path,0,sizeof(path)); /*Input*/ for(i=1;i<=n;i++) { cin>>p[i]>>d[i]; s[i]=p[i]+d[i]; v[i]=p[i]-d[i]; } int fix=m*20; //总修正值,修正极限为从[-400,400]映射到[0,800] /*DP*/ dp[0][fix]=0; //由于修正了数值,因此dp[0][fix]才是真正的dp[0][0] for(j=1;j<=m;j++) for(k=0;k<=2*fix;k++) { if(dp[j-1][k]>=0) //区间已平移,dp[0][fix]才是真正的dp[0][0] { for(i=1;i<=n;i++) if(dp[j][ k+v[i] ] < dp[j-1][k]+s[i]) { if(select(j-1,k,i,v)) { dp[j][ k+v[i] ] = dp[j-1][k]+s[i]; path[j][ k+v[i] ] = i; } } } } /*Output*/ for(k=0;k<=fix;k++) if(dp[m][fix-k]>=0 || dp[m][fix+k]>=0) //从中间向两边搜索最小辨控差的位置k break; int div=dp[m][fix-k] > dp[m][fix+k] ? (fix-k):(fix+k); //最小辨控差 cout<<"Jury #"<<time++<<endl; cout<<"Best jury has value "; //辩方总值 = (辨控和+辨控差+修正值)/2 cout<<(dp[m][div]+div-fix)/2<<" for prosecution and value "; //控方总值 = (辨控和-辨控差+修正值)/2 cout<<(dp[m][div]-div+fix)/2<<" for defence:"<<endl; int* id=new int[m]; for(i=0,j=m,k=div;i<m;i++) { id[i]=path[j][k]; k-=v[ id[i] ]; j--; } sort(id,id+m); //升序输出候选人编号 for(i=0;i<m;i++) cout<<‘ ‘<<id[i]; cout<<endl<<endl; } return 0;}
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。