目录
poj1456
题目链接
http://poj.org/problem?id=1456
题目
有n个商品,每个商品有利润p和过期时间d,每天只能卖一个商品。过期的商品不能再卖,问如何安排每天卖的商品。
题解
首先,用一个小根堆来维护所有商品的利润p。
- 将所有商品按过期时间排序;
- 从头到尾扫描商品,如果该商品的过期时间t=堆的大小,则比较它与堆顶的大小,如果它大于堆顶的话,那么堆顶弹出,再将该商品加入堆;如果该商品的过期时间>堆的大小,直接push进去即可;(为什么不讨论<的情况呢,因为根本不会小于)
- 对堆里所有元素求和,结果就是答案。
注意:priority_queue默认的是大根堆!!!!!!!!!
代码
#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <queue>
#include <stack>
#include <algorithm>
#include <map>
#include <set>
#include <vector>
#include <cmath>
using namespace std;
#define INIT(x) memset(x,0,sizeof(x))
#define eps 1e-8
#define next next_
typedef long long ll;
typedef unsigned long long ull;
const int INF = 0x7fffffff;
const int inf = 0x3f3f3f3f;
const int maxn = 200005;
const int N = 105;
inline void read(int &x) {
int f=1;x=0;char s=getchar();
while(s<'0'||s>'9'){if(s=='-')f=-1;s=getchar();}
while(s>='0'&&s<='9'){x=x*10+s-'0';s=getchar();}
x*=f;
}
/*inline void caltime(int tt) {
tt = clock() - tt;
cerr << (double)tt/CLOCKS_PER_SEC << " seconds!" << endl;
}*/
inline void print_matrix(int a[N][N],int n,int m,bool flag=0) {
if(flag) {
for(int i=1;i<=n;i++) {
for(int j=1;j<=m;j++)
cout<<a[i][j]<<" ";
cout<<endl;
}
}else {
for(int i=0;i<n;i++) {
for(int j=0;j<m;j++)
cout<<a[i][j]<<" ";
cout<<endl;
}
}
}
inline void print_line(int *a,int n,bool flag=0) {
if(flag) {
for(int i=1;i<=n;i++)
cout<<a[i]<<" ";
}else {
for(int i=0;i<n;i++)
cout<<a[i]<<" ";
}
cout<<endl;
}
int n;
struct node{
int p,d;
bool operator<(const node&b)const{
return d<b.d;
}
}a[maxn];
priority_queue<int,vector<int>,greater<int> > q; //优先队列默认是大根堆!!!
ll get_sum() {
ll res = 0;
while(!q.empty()) {
res += q.top();
q.pop();
}
return res;
}
int main()
{
while(cin>>n) {
for(int i=0;i<n;i++) {
read(a[i].p),read(a[i].d);
}
sort(a,a+n);
for(int i=0;i<n;i++) {
if(a[i].d>q.size()) {
q.push(a[i].p);
}else if(a[i].d==q.size()) {
if(q.top()<a[i].p) {
q.pop();
q.push(a[i].p);
}
}
}
ll ans = get_sum();
cout<<ans<<endl;
}
return 0;
}
poj2422
题目链接
http://poj.org/problem?id=1456
题目
给定M个长度为N的序列,从每个序列中任取一个数求和,可以构成个和,求其中最小的N个和。
题解
首先将M组个数全部排序;
假设M=2,那么最小值一定是a[0]+b[0], 次小值就是min(a[0]+b[1],a[1]+b[0]) ,以此类推。
这样我们每次就有了两个扩展方案,但是可能会出现重复,比如a[1]+b[1]可以同时由a[0]+b[1]和a[1]+b[0]扩展而来;
于是就需要加上标记,如果之前扩展过b,那么下一次只能继续扩展b(可以模拟一下);
这一步我们通过一个三元组(i,j,flag)来得到,其中flag表示之前是否扩展过b。
现在扩展到M组,扩展完前两组后,把新得到的N个数作为一组,和第三组去比较,依次类推,总共会生成M-1次。
代码
#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <queue>
#include <stack>
#include <algorithm>
#include <map>
#include <set>
#include <vector>
#include <cmath>
using namespace std;
#define INIT(x) memset(x,0,sizeof(x))
#define eps 1e-8
#define next next_
typedef long long ll;
typedef unsigned long long ull;
const int INF = 0x7fffffff;
const int inf = 0x3f3f3f3f;
const int maxn = 3005;
const int N = 105;
inline void read(int &x) {
int f=1;x=0;char s=getchar();
while(s<'0'||s>'9'){if(s=='-')f=-1;s=getchar();}
while(s>='0'&&s<='9'){x=x*10+s-'0';s=getchar();}
x*=f;
}
/*inline void caltime(int tt) {
tt = clock() - tt;
cerr << (double)tt/CLOCKS_PER_SEC << " seconds!" << endl;
}*/
inline void print_matrix(int a[N][N],int n,int m,bool flag=0) {
if(flag) {
for(int i=1;i<=n;i++) {
for(int j=1;j<=m;j++)
cout<<a[i][j]<<" ";
cout<<endl;
}
}else {
for(int i=0;i<n;i++) {
for(int j=0;j<m;j++)
cout<<a[i][j]<<" ";
cout<<endl;
}
}
}
inline void print_line(int *a,int n,bool flag=0) {
if(flag) {
for(int i=1;i<=n;i++)
cout<<a[i]<<" ";
}else {
for(int i=0;i<n;i++)
cout<<a[i]<<" ";
}
cout<<endl;
}
int t,n,m,f[maxn];
vector<int> e[maxn];
struct node {
int i,j,sum;
bool flag;
bool operator<(const node&b) const{
return sum>b.sum;
}
};
priority_queue<node> q;
void solve(int k) {
int cnt=0;
while(!q.empty()) {
node temp = q.top();
q.pop();
f[cnt++] = temp.sum;
if(temp.j+1<n) q.push({temp.i,temp.j+1,e[k-1][temp.i]+e[k][temp.j+1],1}); //注意不能超出范围
if(!temp.flag&&temp.i+1<n) q.push({temp.i+1,temp.j,e[k-1][temp.i+1]+e[k][temp.j],0});
if(cnt==n) break;
}
for(int i=0;i<n;i++) {
e[k][i] = f[i];
}
}
void init() {while(!q.empty()) q.pop();}
int main()
{
cin>>t;
while(t--) {
read(m),read(n);
int x;
for(int i=0;i<m;i++) {
e[i].clear();
for(int j=0;j<n;j++) {
read(x);
e[i].push_back(x);
}
sort(e[i].begin(),e[i].end());
}
for(int i=0;i<n;i++) f[i] = e[0][i];
for(int i=1;i<m;i++) {
init();
q.push({0,0,e[i-1][0]+e[i][0],0});
solve(i);
}
for(int i=0;i<n;i++) {
if(i==n-1) cout<<f[i];
else cout<<f[i]<<" ";
}
cout<<endl;
}
return 0;
}