hdu--4435--charge-station+貪心

acm_lkl發表於2020-04-04

題目連結:點選進入
首先按照題目的規則,肯定在編號越小的點建立加油站比編號大的點建立加油站要好,但是我們還是不能一下就知道哪些點應該建立加油站。好在點比較少,所以我們可以先在所有的點上都建好加油站,然後再從編號最大的點開始嘗試著刪除。這樣剩下的問題就是判斷當前的情況下能不能滿足題目要求,其實我們可以將建有加油站的點都看成源點,那麼就是一個多源點的最短路問題了,我們只需要計算出每個點到最近源點的距離,然後再與題目中給定的距離D進行比較就好了;當然還要保證可以從第一個源點到達另外的源點。

程式碼如下:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<cmath>
using namespace std;

#define INF 0x3f3f3f3f
const int maxn=220;
int dist[maxn][maxn],flag[maxn];
int dx[maxn],dy[maxn],D,n;

int cal(int i,int j)
{
    double x=(dx[i]-dx[j])*(dx[i]-dx[j])*1.0;
    double y=(dy[i]-dy[j])*(dy[i]-dy[j])*1.0;
    return ceil(sqrt(x+y));
}

int check()
{
    int vis[maxn];
    int d[maxn];
    queue<int>q;
    memset(vis,0,sizeof(vis));
    vis[0]=1;
    for(int i=0;i<n;i++)
    {
        if(flag[i])
            d[i]=0;
        else
            d[i]=INF;
    }
    q.push(0);
    while(!q.empty())
    {
        int cur=q.front();
        q.pop();
        for(int i=0;i<n;i++)
        {
            if(!vis[i]&&dist[cur][i]<=D)
            {
                if(flag[i])
                {
                    vis[i]=1;
                    q.push(i);
                }
                d[i]=min(d[i],dist[cur][i]+d[cur]);
            }
        }
    }
    for(int i=0;i<n;i++)
    {
       if(flag[i]&&!vis[i]) return 0;
       if(!flag[i]&&d[i]>D/2) return 0;
    }
    return 1;
}

int main()
{
    //freopen("in.txt","r",stdin);
    while(scanf("%d%d",&n,&D)!=EOF)
    {
        for(int i=0;i<n;i++)
            scanf("%d%d",&dx[i],&dy[i]);
        for(int i=0;i<n;i++)
            for(int j=i+1;j<n;j++)
               dist[i][j]=dist[j][i]=cal(i,j);
        for(int i=0;i<n;i++)
            flag[i]=1;
        if(!check())
            printf("-1\n");
        else
        {
            for(int i=n-1;i>0;i--)
            {
                flag[i]=0;
                if(!check())
                  flag[i]=1;
            }
            int j;
            for(j=n-1;j>=0;j--)
                if(flag[j]) break;
            for(int i=j;i>=0;i--)
                printf("%d",flag[i]);
            printf("\n");
        }
    }
  return 0;
}