HDU-6396:Swordsman(优先队列)

Swordsman
Time Limit: 2000/1000 MS (Java/Others)
Memory Limit: 131072/131072 K (Java/Others)

Problem Description
Lawson is a magic swordsman with k kinds of magic attributes v 1 , v 2 , v 3 , , v k . Now Lawson is faced with n monsters and the i-th monster also has k kinds of defensive attributes a i , 1 , a i , 2 , a i , 3 , , a i , k . If v 1 a i , 1 and v 2 a i , 2 and v 3 a i , 3 and … and v k a i , k , Lawson can kill the i-th monster (each monster can be killed for at most one time) and get EXP from the battle, which means v j will increase b i , j for j = 1 , 2 , 3 , , k .
Now we want to know how many monsters Lawson can kill at most and how much Lawson’s magic attributes can be maximized.

Input
There are multiple test cases. The first line of input contains an integer T , indicating the number of test cases. For each test case:
The first line has two integers n and k ( 1 n 10 5 , 1 k 5 ) .
The second line has k non-negative integers (initial magic attributes) v 1 , v 2 , v 3 , , v k .
For the next n lines, the i-th line contains 2 k non-negative integers a i , 1 , a i , 2 , a i , 3 , , a i , k , b i , 1 , b i , 2 , b i , 3 , , b i , k .
It’s guaranteed that all input integers are no more than 10 9 and v j + i = 1 n b i , j 10 9 for j = 1 , 2 , 3 , , k .

It is guaranteed that the sum of all n 5 × 10 5 .
The input data is very large so fast IO (like fread) is recommended.

Output
For each test case:
The first line has one integer which means the maximum number of monsters that can be killed by Lawson.
The second line has k integers v 1 , v 2 , v 3 , , v k and the i-th integer means maximum of the i-th magic attibute.

Sample Input
1
4 3
7 1 1
5 5 2 6 3 1
24 1 1 1 2 1
0 4 1 5 1 1
6 0 1 5 3 1

Sample Output
3
23 8 4

Hint

For the sample, initial V = [7, 1, 1]
① kill monster #4 (6, 0, 1), V + [5, 3, 1] = [12, 4, 2]
② kill monster #3 (0, 4, 1), V + [5, 1, 1] = [17, 5, 3]
③ kill monster #1 (5, 5, 2), V + [6, 3, 1] = [23, 8, 4]
After three battles, Lawson are still not able to kill monster #2 (24, 1, 1)
because 23 < 24.

思路:我们可以不断从这 n 个怪物中筛选出满足条件的。
1. 把满足 a [ 1 ] v [ 1 ] 的怪物放入一个优先队列 P 2
2. 然后从 P 2 中选出所有满足 a [ 2 ] v [ 2 ] 的怪物放入一个优先队列 P 3
3. 然后从 P 3 中选出所有满足 a [ 3 ] v [ 3 ] 的怪物放入一个优先队列 P 4
4. 然后从 P 4 中选出所有满足 a [ 4 ] v [ 4 ] 的怪物放入一个优先队列 P 5
5. 然后从 P 5 中选出所有满足 a [ 5 ] v [ 5 ] 的怪物,则这些怪物是可以被杀死的,把杀死这个怪物可以获得的属性更新至 v 数组,然后重复 1 操作。
不过我代码写得很冗长。。题解代码非常简洁
这题还要用快速读入,自己的快读T了,用dls的就A了%%%。。
PS:用dls的快读手动测不了数据,必须用文件输入输出

#include<bits/stdc++.h>
using namespace std;
const int MAX=1e5+10;
const int MOD=1e9+7;
const int INF=1e9+7;
const double PI=acos(-1.0);
typedef long long ll;
namespace IO{
    #define BUF_SIZE 100000
    #define OUT_SIZE 100000
    #define ll long long
    //fread->read

    bool IOerror=0;
    inline char nc(){
        static char buf[BUF_SIZE],*p1=buf+BUF_SIZE,*pend=buf+BUF_SIZE;
        if (p1==pend){
            p1=buf; pend=buf+fread(buf,1,BUF_SIZE,stdin);
            if (pend==p1){IOerror=1;return -1;}
            //{printf("IO error!\n");system("pause");for (;;);exit(0);}
        }
        return *p1++;
    }
    inline bool blank(char ch){return ch==' '||ch=='\n'||ch=='\r'||ch=='\t';}
    inline void read(int &x){
        bool sign=0; char ch=nc(); x=0;
        for (;blank(ch);ch=nc());
        if (IOerror)return;
        if (ch=='-')sign=1,ch=nc();
        for (;ch>='0'&&ch<='9';ch=nc())x=x*10+ch-'0';
        if (sign)x=-x;
    }
    inline void read(ll &x){
        bool sign=0; char ch=nc(); x=0;
        for (;blank(ch);ch=nc());
        if (IOerror)return;
        if (ch=='-')sign=1,ch=nc();
        for (;ch>='0'&&ch<='9';ch=nc())x=x*10+ch-'0';
        if (sign)x=-x;
    }
    inline void read(double &x){
        bool sign=0; char ch=nc(); x=0;
        for (;blank(ch);ch=nc());
        if (IOerror)return;
        if (ch=='-')sign=1,ch=nc();
        for (;ch>='0'&&ch<='9';ch=nc())x=x*10+ch-'0';
        if (ch=='.'){
            double tmp=1; ch=nc();
            for (;ch>='0'&&ch<='9';ch=nc())tmp/=10.0,x+=tmp*(ch-'0');
        }
        if (sign)x=-x;
    }
    inline void read(char *s){
        char ch=nc();
        for (;blank(ch);ch=nc());
        if (IOerror)return;
        for (;!blank(ch)&&!IOerror;ch=nc())*s++=ch;
        *s=0;
    }
    inline void read(char &c){
        for (c=nc();blank(c);c=nc());
        if (IOerror){c=-1;return;}
    }
    //fwrite->write
    struct Ostream_fwrite{
        char *buf,*p1,*pend;
        Ostream_fwrite(){buf=new char[BUF_SIZE];p1=buf;pend=buf+BUF_SIZE;}
        void out(char ch){
            if (p1==pend){
                fwrite(buf,1,BUF_SIZE,stdout);p1=buf;
            }
            *p1++=ch;
        }
        void print(int x){
            static char s[15],*s1;s1=s;
            if (!x)*s1++='0';if (x<0)out('-'),x=-x;
            while(x)*s1++=x%10+'0',x/=10;
            while(s1--!=s)out(*s1);
        }
        void println(int x){
            static char s[15],*s1;s1=s;
            if (!x)*s1++='0';if (x<0)out('-'),x=-x;
            while(x)*s1++=x%10+'0',x/=10;
            while(s1--!=s)out(*s1); out('\n');
        }
        void print(ll x){
            static char s[25],*s1;s1=s;
            if (!x)*s1++='0';if (x<0)out('-'),x=-x;
            while(x)*s1++=x%10+'0',x/=10;
            while(s1--!=s)out(*s1);
        }
        void println(ll x){
            static char s[25],*s1;s1=s;
            if (!x)*s1++='0';if (x<0)out('-'),x=-x;
            while(x)*s1++=x%10+'0',x/=10;
            while(s1--!=s)out(*s1); out('\n');
        }
        void print(double x,int y){
            static ll mul[]={1,10,100,1000,10000,100000,1000000,10000000,100000000,
                1000000000,10000000000LL,100000000000LL,1000000000000LL,10000000000000LL,
                100000000000000LL,1000000000000000LL,10000000000000000LL,100000000000000000LL};
            if (x<-1e-12)out('-'),x=-x;x*=mul[y];
            ll x1=(ll)floor(x); if (x-floor(x)>=0.5)++x1;
            ll x2=x1/mul[y],x3=x1-x2*mul[y]; print(x2);
            if (y>0){out('.'); for (size_t i=1;i<y&&x3*mul[i]<mul[y];out('0'),++i); print(x3);}
        }
        void println(double x,int y){print(x,y);out('\n');}
        void print(char *s){while (*s)out(*s++);}
        void println(char *s){while (*s)out(*s++);out('\n');}
        void flush(){if (p1!=buf){fwrite(buf,1,p1-buf,stdout);p1=buf;}}
        ~Ostream_fwrite(){flush();}
    }Ostream;
    inline void print(int x){Ostream.print(x);}
    inline void println(int x){Ostream.println(x);}
    inline void print(char x){Ostream.out(x);}
    inline void println(char x){Ostream.out(x);Ostream.out('\n');}
    inline void print(ll x){Ostream.print(x);}
    inline void println(ll x){Ostream.println(x);}
    inline void print(double x,int y){Ostream.print(x,y);}
    inline void println(double x,int y){Ostream.println(x,y);}
    inline void print(char *s){Ostream.print(s);}
    inline void println(char *s){Ostream.println(s);}
    inline void println(){Ostream.out('\n');}
    inline void flush(){Ostream.flush();}
    #undef ll
    #undef OUT_SIZE
    #undef BUF_SIZE
};
int v[6],k;
struct lenka{int a[6],b[6];}a[MAX];
int cmp(const lenka& A,const lenka& B){return A.a[1]<B.a[1];}
struct lenka2{int a[6],b[6];bool operator<(const lenka2 A)const{return A.a[2]<a[2];}};
struct lenka3{int a[6],b[6];bool operator<(const lenka3 A)const{return A.a[3]<a[3];}};
struct lenka4{int a[6],b[6];bool operator<(const lenka4 A)const{return A.a[4]<a[4];}};
struct lenka5{int a[6],b[6];bool operator<(const lenka5 A)const{return A.a[5]<a[5];}};
lenka2 to2(const lenka& B)
{
    lenka2 A;
    for(int i=1;i<=k;i++)A.a[i]=B.a[i];
    for(int i=1;i<=k;i++)A.b[i]=B.b[i];
    return A;
}
lenka3 to3(const lenka2& B)
{
    lenka3 A;
    for(int i=1;i<=k;i++)A.a[i]=B.a[i];
    for(int i=1;i<=k;i++)A.b[i]=B.b[i];
    return A;
}
lenka4 to4(const lenka3& B)
{
    lenka4 A;
    for(int i=1;i<=k;i++)A.a[i]=B.a[i];
    for(int i=1;i<=k;i++)A.b[i]=B.b[i];
    return A;
}
lenka5 to5(const lenka4& B)
{
    lenka5 A;
    for(int i=1;i<=k;i++)A.a[i]=B.a[i];
    for(int i=1;i<=k;i++)A.b[i]=B.b[i];
    return A;
}
void add(const lenka &A){for(int i=1;i<=k;i++)v[i]+=A.b[i];}
void add(const lenka2 &A){for(int i=1;i<=k;i++)v[i]+=A.b[i];}
void add(const lenka3 &A){for(int i=1;i<=k;i++)v[i]+=A.b[i];}
void add(const lenka4 &A){for(int i=1;i<=k;i++)v[i]+=A.b[i];}
void add(const lenka5 &A){for(int i=1;i<=k;i++)v[i]+=A.b[i];}
priority_queue<lenka2>p2;
priority_queue<lenka3>p3;
priority_queue<lenka4>p4;
priority_queue<lenka5>p5;
int ans;
void cal()
{
    int tag=1;
    while(tag)
    {
        tag=0;
        if(k==1)continue;
        while(!p2.empty()&&p2.top().a[2]<=v[2])
        {
            if(k==2)ans++,add(p2.top()),tag=1;
            else p3.push(to3(p2.top()));
            p2.pop();
        }
        if(k==2)continue;
        while(!p3.empty()&&p3.top().a[3]<=v[3])
        {
            if(k==3)ans++,add(p3.top()),tag=1;
            else p4.push(to4(p3.top()));
            p3.pop();
        }
        if(k==3)continue;
        while(!p4.empty()&&p4.top().a[4]<=v[4])
        {
            if(k==4)ans++,add(p4.top()),tag=1;
            else p5.push(to5(p4.top()));
            p4.pop();
        }
        if(k==4)continue;
        while(!p5.empty()&&p5.top().a[5]<=v[5])
        {
            ans++;
            add(p5.top());
            p5.pop();
            tag=1;
        }
    }
}
int main()
{
    int T;
    IO::read(T);
    while(T--)
    {
        int n;
        IO::read(n);
        IO::read(k);
        for(int i=1;i<=k;i++)IO::read(v[i]);
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=k;j++)IO::read(a[i].a[j]);
            for(int j=1;j<=k;j++)IO::read(a[i].b[j]);
        }
        sort(a+1,a+n+1,cmp);
        ans=0;
        for(int i=1;i<=n;i++)
        {
            cal();
            if(a[i].a[1]<=v[1])
            {
                if(k==1)ans++,add(a[i]);
                else p2.push(to2(a[i]));
            }
        }
        cal();
        printf("%d\n",ans);
        for(int i=1;i<=k;i++)printf("%d%c",v[i],i==k?'\n':' ');
        while(!p2.empty())p2.pop();
        while(!p3.empty())p3.pop();
        while(!p4.empty())p4.pop();
        while(!p5.empty())p5.pop();
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/Mitsuha_/article/details/81635572