[P1220] 가로등 끄기

원제 링크
DP 문제인데 검색어도 잘 넘어가요.
그냥 폭력 수색해서 국경까지 가서 불을 끄는 거예요.
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#define LL long long
#define MOD 1000000007
using namespace std;

int ans=9999999,p[55],w[55],tot,n,c;
bool vis[55];

void dfs(int pos,int cost,int cnt,int all)
{
    if(cost>ans) return;
    if(cnt>=n)
    {
        ans=cost;
        return;
    }

    int l,r;
    l=pos;r=pos;
    while(vis[l]&&l>=1) l--;
    while(vis[r]&&r<=n) r++;

    if(l>=1)
    {
        vis[l]=1;
        dfs(l,cost+all*(p[pos]-p[l]),cnt+1,all-w[l]);
        vis[l]=0;
    }

    if(r<=n)
    {
        vis[r]=1;
        dfs(r,cost+all*(p[r]-p[pos]),cnt+1,all-w[r]);
        vis[r]=0;
    }
}

int main()
{
    int i;

    scanf("%d%d",&n,&c);

    for(i=1;i<=n;i++)
    {
        scanf("%d%d",&p[i],&w[i]);
        tot+=w[i];      
    }

    tot-=w[c];
    vis[c]=1;
    dfs(c,0,1,tot);

    printf("%d",ans);
    return 0;
}

좋은 웹페이지 즐겨찾기