codeforces 912 D. Fishes

题目链接

题意：有一个n×m的鱼塘，有一张r×r的渔网，现在往池塘里面放k条鱼(每个格子只能放一条鱼)， 现在撒网的地方是随机的(必须在池塘内)，问能捕的鱼的期望值最大是多少？

思路：BFS+优先队列

感觉很多题都是这么搞得，bfs加优先队列，然后队列里面存的都是结构体，需要重载运算符。

这题也是，只要挑选出前k个贡献最大的点就行了，判断每个点是否计算过，需要用map和pair，不能用vis二维数组标记，因为会超内存，稍微复杂的地方就是根据点的坐标x，y和网的边长计算点的贡献，具体看代码。

#include<bits/stdc++.h>
#define mem(a,b) memset((a),b,sizeof(a))
typedef long long ll;
const int N=100010;
using namespace std;
ll n,m,r,k;
struct node{
    ll x,y;
    ll v;
    bool operator < (const node& b)const{
        return v<b.v;
    }
};
priority_queue<node> q;
ll x,y,ans;
map<pair<int,int>,int> mp;
ll getv(node p)
{
    return (min(n+1,p.x+r)-(max((ll)1,p.x-r+1)+r)+1)*(min(m+1,p.y+r)-(max((ll)1,p.y-r+1)+r)+1);
}
bool ok(node p)
{
    ll xx=p.x,yy=p.y;
    if(mp[make_pair(xx,yy)]==0)
        mp[make_pair(xx,yy)]++;
    else
        return false;
    if(xx<1||xx>n)
        return false;
    if(yy<1&&yy>m)
        return false;
    return true;
}
void bfs()
{
    node p;
    p.x=x;p.y=y;p.v=getv(p);
    if(ok(p))
        q.push(p);
    while(!q.empty())
    {
        int f=0;
        while(f==0)
        {
            p=q.top();
            q.pop();
            //cout<<"&& "<<p.x<<"  "<<p.y<<"  "<<p.v<<"  ";//<<k<<" ";//endl<<endl;
            ans+=p.v;
            k--;
            //cout<<k<<"  "<<ans<<endl<<endl;
            if(k==0) return ;
            node pp;
            pp.x=p.x-1;pp.y=p.y;
            if(ok(pp))
                {pp.v=getv(pp);q.push(pp);f++;}
            pp.x=p.x+1;pp.y=p.y;
            if(ok(pp))
                {pp.v=getv(pp);q.push(pp);f++;}
            pp.x=p.x;pp.y=p.y-1;
            if(ok(pp))
                {pp.v=getv(pp);q.push(pp);f++;}
            pp.x=p.x;pp.y=p.y+1;
            if(ok(pp))
                {pp.v=getv(pp);q.push(pp);f++;}
        }
    }
}
int main()
{
    scanf("%I64d%I64d%I64d%I64d",&n,&m,&r,&k);
    x=(n+1)/2;y=(m+1)/2;
    ans=0;
    bfs();
    //cout<<"%%"<<ans<<" "<<(m-r+1)*(n-r+1)<<endl;
    double anss=ans*1.0/((m-r+1)*(n-r+1));
    printf("%.9f",anss);
    return 0;
}