UVa 10344 - 23 out of 5

2442 단어 최적화permutation
TLE 를 역추적하기 시작했는데 나중에 전체 정렬 + DFS AC 로 변경되었습니다.
코드는 다음과 같습니다.
(0.728s):
#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <cmath>
using namespace std;
int a[5], flag;
void solve(int cur, int sum)
{
    if(cur == 5)
    {
        if(sum == 23)
            flag = 1;
        return ;
    }
    solve(cur + 1, sum + a[cur]);
    if(flag)
        return;
    solve(cur + 1, sum - a[cur]);
    if(flag)
        return;
    solve(cur + 1, sum * a[cur]);
    if(flag)
        return;
}
int main()
{
#ifdef test
    freopen("sample.txt", "r", stdin);
#endif
    int b[5], s[5];
    while(1)
    {
        for(int i = 0; i < 5; i++)
            scanf("%d", &b[i]);
        if(!b[0])
            break;
        for(int i = 0; i < 5; i++)
            s[i] = i;
        flag = 0;
        do
        {
            for(int i = 0; i < 5; i++)
                a[s[i]] = b[i];
            solve(1, a[0]);
            if(flag)
                break;
        }
        while(next_permutation(s, s + 5));
        if(flag)
            printf("Possible
"); else printf("Impossible
"); } return 0; }

정렬 후 처리가 더욱 간단해지고 최적화됩니다(0.432s).
#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <cmath>
using namespace std;
int a[5], flag;
void solve(int cur, int sum)
{
    if(cur == 5)
    {
        if(sum == 23)
            flag = 1;
        return ;
    }
    solve(cur + 1, sum + a[cur]);
    if(flag)
        return;
    solve(cur + 1, sum - a[cur]);
    if(flag)
        return;
    solve(cur + 1, sum * a[cur]);
    if(flag)
        return;
}
int main()
{
#ifdef test
    freopen("sample.txt", "r", stdin);
#endif
    while(1)
    {
        for(int i = 0; i < 5; i++)
            scanf("%d", &a[i]);
        if(!a[0])
            break;
        flag = 0;
        sort(a, a + 5); //     
        do
        {
            solve(1, a[0]);
            if(flag)
                break;
        }
        while(next_permutation(a, a + 5));
        if(flag)
            printf("Possible
"); else printf("Impossible
"); } return 0; }

좋은 웹페이지 즐겨찾기