题意
传送门 POJ 3370
题解
组合数问题,求组合数模 c c c 等于 0 0 0 的方案。首先想到 D P DP DP,配合 b i t s e t bitset bitset 优化时间复杂度应该可行;然后一看输出,不仅需要判断是否存在合理方案,还要输出路径;此路不通。
观察 c ≤ n c\leq n c≤n,且模 c c c 运算取值区间为 [ 0 , c − 1 ] [0,c-1] [0,c−1],根据抽屉原理, a a a 模 c c c 运算的前缀和(考虑空集)必定会出现重复的情况。假设前缀和 s u m [ i ] = s u m [ j ] sum[i]=sum[j] sum[i]=sum[j],此时对应的区间 ( i , j ] (i,j] (i,j] 模 c c c 等于 0 0 0,必定存在答案且答案为 a a 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;
}