POJ 3370 抽屉原理

题意

传送门 POJ 3370

题解

组合数问题，求组合数模 $c$ 等于 $0$ 的方案。首先想到 $DP$，配合 $bitset$ 优化时间复杂度应该可行；然后一看输出，不仅需要判断是否存在合理方案，还要输出路径；此路不通。

观察 $c\le n$，且模 $c$ 运算取值区间为 $[0,c-1]$，根据抽屉原理，$a$ 模 $c$ 运算的前缀和（考虑空集）必定会出现重复的情况。假设前缀和 $sum[i]=sum[j]$，此时对应的区间 $(i,j]$ 模 $c$ 等于 $0$，必定存在答案且答案为 $a$ 的连续子数组。

#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstring>
using namespace std;
#define inf 0x3f3f3f3f
#define maxn 100005
int c, n, a[maxn], rec[maxn];

int main()
{
    while (~scanf("%d%d", &c, &n) && (c | n))
    {
        for (int i = 1; i <= n; i++) scanf("%d", a + i);
        memset(rec, -1, sizeof(rec));
        int s = 0;
        rec[s] = 0;
        for (int i = 1; i <= n; i++)
        {
            s = (s + a[i]) % c;
            if (rec[s] != -1)
            {
                for (int j = rec[s] + 1; j <= i; j++)
                {
                    printf("%d%c", j, j == i ? '\n' : ' ');
                }
                break;
            }
            rec[s] = i;
        }
    }
    return 0;
}