项链（2019牛客寒假算法基础集训营 Day6-C）

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/u011815404/article/details/88382320

【题目描述】

小B想给她的新项链染色。
现在有m种颜色，对于第i种颜色，小B有a_i单位的颜料，每单位颜料可以染项链的一个珠子；
同时，小B对于第i种颜色的喜爱度为b_i。
已知项链有n个珠子，求染色后每个珠子的颜色的喜爱度之和的最大值。
(每个珠子只能至多被染一次，不被染色则喜爱度为0)    

【输入描述】

第一行两个数n,m
第二行m个数a_i
第三行m个数b_i

【输出描述】

一个数表示答案

【样例】

示例1

输入
5 3
1 2 3
3 2 1
输出
9

示例2

输入
5 3
1 2 1
3 2 1
输出
9

思路：使用结构体，在颜色数量足够的情况下，按照喜爱程度贪心

【源代码】

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<string>
#include<cstring>
#include<cmath>
#include<ctime>
#include<algorithm>
#include<utility>
#include<stack>
#include<queue>
#include<vector>
#include<set>
#include<map>
#define PI acos(-1.0)
#define E 1e-9
#define INF 0x3f3f3f3f
#define LL long long
const int MOD=1E9+7;
const int N=5000000+5;
const int dx[]= {-1,1,0,0};
const int dy[]= {0,0,-1,1};
using namespace std;

struct Node{
	LL a;
	LL b;
	bool operator <(Node &rhs)const{
	    return b>rhs.b;
	}
}node[N];
int main(){
	int n,m;
	cin>>n>>m;
	for(int i=1;i<=m;i++)
        cin>>node[i].a;
	for(int i=1;i<=m;i++)
        cin>>node[i].b;

	sort(node+1,node+m+1);

	LL res=0;
	LL sum=0;
	for(int i=1;i<=m;i++){
		sum+=node[i].a;
		if(sum>n){
			sum=(n-(sum-node[i].a));
			res=res+node[i].b*sum;
			break;
		}
		else
			res=res+node[i].a*node[i].b;
	}
	cout<<res<<endl;

	return 0;
}